Soumic commited on
Commit
1b17226
1 Parent(s): 8819eb2

:zap: Update max-steps, code is ready to fine-tune

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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": 100, # 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,
 
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,