devforfu
commited on
Commit
·
c6b22f5
1
Parent(s):
87ccfbe
Fixed typo; enable prograss bar if using interactive
Browse files- realfake/train.py +2 -1
realfake/train.py
CHANGED
@@ -61,7 +61,8 @@ def prepare_trainer(args: RealFakeParams) -> pl.Trainer:
|
|
61 |
)
|
62 |
|
63 |
if job_id is not None:
|
64 |
-
trainer_params["
|
|
|
65 |
trainer_params["strategy"] = args.accelerator.strategy
|
66 |
|
67 |
return pl.Trainer(**trainer_params)
|
|
|
61 |
)
|
62 |
|
63 |
if job_id is not None:
|
64 |
+
trainer_params["enable_progress_bar"] = True
|
65 |
+
trainer_params["plugins"] = SLURMEnvironment(requeue_signal=signal.SIGHUP)
|
66 |
trainer_params["strategy"] = args.accelerator.strategy
|
67 |
|
68 |
return pl.Trainer(**trainer_params)
|