Add code so loss function uses torch.Size([x,1]) instead of torch.Size([x]) ba9c868 Hannes Kuchelmeister commited on Mar 10, 2022