wangjin2000 commited on
Commit
7f2ddaa
·
verified ·
1 Parent(s): 7f294be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def finetune(base_model_path, peptide_length): #, train_dataset, test_dataset)
108
  gradient_accumulation_steps=2,
109
  push_to_hub=True, #jw 20240918
110
  hub_token = HF_TOKEN, #jw 20240918
111
- dataloader_pin_memory=False #jw 20241119 true for CPU
112
  )
113
 
114
  # Initialize Trainer
 
108
  gradient_accumulation_steps=2,
109
  push_to_hub=True, #jw 20240918
110
  hub_token = HF_TOKEN, #jw 20240918
111
+ dataloader_pin_memory=False, #jw 20241119 true for CPU
112
  )
113
 
114
  # Initialize Trainer