NbAiLabArchive
/

test_NCC_small_pytorch

Inference Endpoints

Model card Files Files and versions Community

pere commited on Dec 3, 2021

Commit

f307d2f

•

1 Parent(s): 923f9ad

fix

Files changed (1) hide show

run_flax.sh +1 -1

run_flax.sh CHANGED Viewed

@@ -11,7 +11,7 @@
     --learning_rate="6e-4" \
     --warmup_steps="10000" \
     --overwrite_output_dir \
-    --num_train_epochs ="10" \
     --adam_beta1="0.9" \
     --adam_beta2="0.98" \
     --logging_steps="10000" \

     --learning_rate="6e-4" \
     --warmup_steps="10000" \
     --overwrite_output_dir \
+    --num_train_epochs="10" \
     --adam_beta1="0.9" \
     --adam_beta2="0.98" \
     --logging_steps="10000" \