Spaces:
Running
Running
Commit
路
44bb427
1
Parent(s):
bfdf9fe
Update hparams.py
Browse files- 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=
|
| 77 |
fps=15,
|
| 78 |
|
| 79 |
-
batch_size=
|
| 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=
|
| 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,
|