Saving weights of epoch 1 at step 92
Browse files
flax_model.msgpack
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1419367919
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:72f9b01147de892ad202a09ca9363f9299fb3bdda34dea308fc38ee984f63547
|
| 3 |
size 1419367919
|
results_tensorboard/events.out.tfevents.1626459390.t1v-n-8cb15980-w-0.890191.3.v2
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:67e74e061f8fd301ab09d304e7b49db5a83c95a10eebeb03adc1fd087bd0b69b
|
| 3 |
+
size 25152
|
src/train.py
CHANGED
|
@@ -73,7 +73,7 @@ def main():
|
|
| 73 |
|
| 74 |
per_device_batch_size=4
|
| 75 |
seed=0
|
| 76 |
-
num_train_epochs=
|
| 77 |
learning_rate=2e-5
|
| 78 |
|
| 79 |
|
|
|
|
| 73 |
|
| 74 |
per_device_batch_size=4
|
| 75 |
seed=0
|
| 76 |
+
num_train_epochs=5
|
| 77 |
learning_rate=2e-5
|
| 78 |
|
| 79 |
|