flax-community
/

gpt2-rap-lyric-generator

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Shankhdhar commited on Jul 16, 2021

Commit

fb89469

•

1 Parent(s): 2ec76c9

new sesh

Files changed (2) hide show

flax_model.msgpack +1 -1
run.sh +2 -3

flax_model.msgpack CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:742e9e9fd4077234f5e0a5f68fd466c23b7ffae51c771abaafa5ae4d083edfca
 size 497764120

 version https://git-lfs.github.com/spec/v1
+oid sha256:f0e9481eef2c6ddc206828356257030bf1b3f354b37ba73272e3ba5352f99dbc
 size 497764120

run.sh CHANGED Viewed

@@ -14,8 +14,7 @@ python3 run_clm_flax.py \
     --learning_rate="5e-4"  \
     --adam_beta1="0.9" --adam_beta2="0.98" --weight_decay="0.01" \
     --overwrite_output_dir \
-    --num_train_epochs="100" \
     --logging_steps="100" \
-    --save_steps="100" \
     --eval_steps="100" \
-    --push_to_hub

     --learning_rate="5e-4"  \
     --adam_beta1="0.9" --adam_beta2="0.98" --weight_decay="0.01" \
     --overwrite_output_dir \
+    --num_train_epochs="60" \
     --logging_steps="100" \
+    --save_steps="200" \
     --eval_steps="100" \