Model:	 lstm 
Number of layers:	 2 
Learning rate:	 0.0001 
Batch size:	 1024 
Training data shape:	 x = torch.Size([6000000, 10, 7]) y = torch.Size([6000000])
Validation data shape:	 x = torch.Size([600000, 10, 7]) y = torch.Size([600000])
Test data shape:	 x = torch.Size([600000, 10, 7]) y = torch.Size([600000])
Trainable parameters:	 29809 
Optimizer:	 Adam (
Parameter Group 0
    amsgrad: False
    betas: (0.9, 0.999)
    eps: 1e-08
    lr: 0.0001
    weight_decay: 0.0
)