Skip to content

Commit

Permalink
cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
albertz committed Nov 27, 2023
1 parent acc4308 commit eb9b05e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion users/zeyer/experiments/exp2023_04_25_rf/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ def train(
for k, v in dict(
log_verbosity=5,
num_epochs=150,
time_rqmt=10, # TODO change back to 80, only temporary
time_rqmt=80,
mem_rqmt=30 if gpu_mem and gpu_mem > 11 else 15,
cpu_rqmt=4 if (not num_processes or num_processes <= 4) else 3,
horovod_num_processes=num_processes, # legacy name but also applies for Torch
Expand Down

0 comments on commit eb9b05e

Please sign in to comment.