rxxnzz commited on
Commit
04080ca
1 Parent(s): 7952e9e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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 = 20
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