Spaces:
Sleeping
Sleeping
salomonsky
commited on
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,
|