devforfu commited on
Commit
c6b22f5
1 Parent(s): 87ccfbe

Fixed typo; enable prograss bar if using interactive

Browse files
Files changed (1) hide show
  1. 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["plugins"] = SLURMEnvironment(requeue_signal=signal.SIGHUP),
 
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)