Spaces:
Running
on
T4
Running
on
T4
Hugo Flores Garcia
commited on
Commit
•
0a036ac
1
Parent(s):
d48dcc4
remove cancerous subprocess call
Browse files- 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)
|