salomonsky commited on
Commit
44bb427
1 Parent(s): bfdf9fe

Update hparams.py

Browse files
Files changed (1) hide show
  1. hparams.py +3 -3
hparams.py CHANGED
@@ -73,13 +73,13 @@ hparams = HParams(
73
  fmax=7600, # To be increased/reduced depending on data.
74
 
75
  ###################### Our training parameters #################################
76
- img_size=96,
77
  fps=15,
78
 
79
- batch_size=12,
80
  initial_learning_rate=1e-4,
81
  nepochs=200000000000000000, ### ctrl + c, stop whenever eval loss is consistently greater than train loss for ~10 epochs
82
- num_workers=12,
83
  checkpoint_interval=3000,
84
  eval_interval=3000,
85
  save_optimizer_state=True,
 
73
  fmax=7600, # To be increased/reduced depending on data.
74
 
75
  ###################### Our training parameters #################################
76
+ img_size=66,
77
  fps=15,
78
 
79
+ batch_size=4,
80
  initial_learning_rate=1e-4,
81
  nepochs=200000000000000000, ### ctrl + c, stop whenever eval loss is consistently greater than train loss for ~10 epochs
82
+ num_workers=4,
83
  checkpoint_interval=3000,
84
  eval_interval=3000,
85
  save_optimizer_state=True,