amitke commited on
Commit
d21c19f
·
1 Parent(s): 889c9ea

learing rate 1e-3

Browse files
Files changed (1) hide show
  1. train.py +1 -1
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-2):
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 ---