Spaces:
Sleeping
Sleeping
Update back/models/launch.py
Browse files- 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
|