Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
| 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,
|