Q-bert commited on
Commit
b5e8200
·
verified ·
1 Parent(s): 3434018

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ def train_stock_model(stock_symbol, start_date, end_date, feature_range=(10, 100
97
  gradient_accumulation_steps=4,
98
  num_train_epochs=epochs,
99
  warmup_steps=5,
100
- save_steps=100,
101
  learning_rate=learning_rate,
102
  fp16=True,
103
  logging_steps=1,
 
97
  gradient_accumulation_steps=4,
98
  num_train_epochs=epochs,
99
  warmup_steps=5,
100
+ save_steps=10,
101
  learning_rate=learning_rate,
102
  fp16=True,
103
  logging_steps=1,