Spaces:

YinuoGuo27
/

test_kdtalker

Running on Zero

YinuoGuo27 commited on Apr 2

Commit

513c6ff

verified ·

1 Parent(s): 84e8e98

Update difpoint/src/utils/hparams.py

Files changed (1) hide show

difpoint/src/utils/hparams.py CHANGED Viewed

@@ -68,7 +68,7 @@ hparams = HParams(
 	batch_size=16,
 	initial_learning_rate=1e-4,
 	nepochs=300000,  ### ctrl + c, stop whenever eval loss is consistently greater than train loss for ~10 epochs
-	num_workers=20,
 	checkpoint_interval=3000,
 	eval_interval=3000,
 	writer_interval=300,

 	batch_size=16,
 	initial_learning_rate=1e-4,
 	nepochs=300000,  ### ctrl + c, stop whenever eval loss is consistently greater than train loss for ~10 epochs
+	num_workers=0,
 	checkpoint_interval=3000,
 	eval_interval=3000,
 	writer_interval=300,