Solved by ilkarman in post #2 
              
              
                You are passing a numpy array into a torch model. You would have to transform train_x: torch.Tensor(train_x) for it to generate an output. 
However, train_x here doesn’t seem to be your batch but the whole dataset right? In which case you may prefer to wrap it with the DataLoader() class: train_x_lo…