Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -129,7 +129,7 @@ def encode_dataset(ds, limit=-1):
|
|
129 |
label_list
|
130 |
)).map(map_example_to_dict)
|
131 |
|
132 |
-
EPOCH =
|
133 |
BATCH_SIZE = 42
|
134 |
LEARNING_RATE = 1e-5
|
135 |
|
|
|
129 |
label_list
|
130 |
)).map(map_example_to_dict)
|
131 |
|
132 |
+
EPOCH = 1
|
133 |
BATCH_SIZE = 42
|
134 |
LEARNING_RATE = 1e-5
|
135 |
|