Spaces:
Runtime error
Runtime error
Soumic
commited on
Commit
•
1b17226
1
Parent(s):
8819eb2
:zap: Update max-steps, code is ready to fine-tune
Browse files
app.py
CHANGED
@@ -243,7 +243,7 @@ def start():
|
|
243 |
args = {
|
244 |
"output_dir": "output_dnabert-6-mqtl_classification",
|
245 |
"num_train_epochs": 1,
|
246 |
-
"max_steps":
|
247 |
# Set the number of steps you expect to train, originally 1000, takes too much time. So I set it to 10 to run faster and check my code/pipeline
|
248 |
"run_name": "laptop_run_dna-bert-6-mqtl_classification", # Override run_name here
|
249 |
"per_device_train_batch_size": 1,
|
|
|
243 |
args = {
|
244 |
"output_dir": "output_dnabert-6-mqtl_classification",
|
245 |
"num_train_epochs": 1,
|
246 |
+
"max_steps": 20_000, # train 36k + val 4k = 40k
|
247 |
# Set the number of steps you expect to train, originally 1000, takes too much time. So I set it to 10 to run faster and check my code/pipeline
|
248 |
"run_name": "laptop_run_dna-bert-6-mqtl_classification", # Override run_name here
|
249 |
"per_device_train_batch_size": 1,
|