Hugo Flores Garcia commited on
Commit
0a036ac
1 Parent(s): d48dcc4

remove cancerous subprocess call

Browse files
Files changed (1) hide show
  1. scripts/exp/train.py +0 -2
scripts/exp/train.py CHANGED
@@ -413,8 +413,6 @@ def train(
413
  output["other/learning_rate"] = optimizer.param_groups[0]["lr"]
414
  output["other/batch_size"] = z.shape[0]
415
 
416
- output.update(get_gpu_memory_map())
417
-
418
  if (
419
  (engine.state.iteration % grad_acc_steps == 0)
420
  or (engine.state.iteration % epoch_length == 0)
 
413
  output["other/learning_rate"] = optimizer.param_groups[0]["lr"]
414
  output["other/batch_size"] = z.shape[0]
415
 
 
 
416
  if (
417
  (engine.state.iteration % grad_acc_steps == 0)
418
  or (engine.state.iteration % epoch_length == 0)