yderre-aubay commited on
Commit
d5f5022
·
verified ·
1 Parent(s): e1c1d9c

Update back/models/launch.py

Browse files
Files changed (1) hide show
  1. back/models/launch.py +1 -1
back/models/launch.py CHANGED
@@ -364,7 +364,7 @@ def train_model(
364
  checkpoint_model_file = (
365
  f"{output_dir}/checkpoints/model"
366
  )
367
- history_file = f"{output_dir}/history/model
368
 
369
  print(tf.config.list_physical_devices("GPU"))
370
  gpu_avail = len(tf.config.list_physical_devices("GPU")) # True/False
 
364
  checkpoint_model_file = (
365
  f"{output_dir}/checkpoints/model"
366
  )
367
+ history_file = f"{output_dir}/history/model"
368
 
369
  print(tf.config.list_physical_devices("GPU"))
370
  gpu_avail = len(tf.config.list_physical_devices("GPU")) # True/False