Spaces:
Sleeping
Sleeping
amitke commited on
Commit ·
d21c19f
1
Parent(s): 889c9ea
learing rate 1e-3
Browse files
train.py
CHANGED
|
@@ -44,7 +44,7 @@ def to_tensor_loader(X, y, batch_size=32):
|
|
| 44 |
return ds
|
| 45 |
|
| 46 |
def train(symbol: str, seq_len: int = 60, epochs: int = 5, batch_size: int = 32,
|
| 47 |
-
start: str = None, end: str = None, lr: float = 1e-
|
| 48 |
device = torch.device("cpu")
|
| 49 |
|
| 50 |
# --- data ---
|
|
|
|
| 44 |
return ds
|
| 45 |
|
| 46 |
def train(symbol: str, seq_len: int = 60, epochs: int = 5, batch_size: int = 32,
|
| 47 |
+
start: str = None, end: str = None, lr: float = 1e-3):
|
| 48 |
device = torch.device("cpu")
|
| 49 |
|
| 50 |
# --- data ---
|