Yizhi Wang

> ValueError: Unexpected behavior when reshaping between beam width and batch size. The reshaped tensor has shape: (1, 5, 256). We expected it to have shape (batch_size, beam_width, depth) ==...

I solved the problem by setting the 'drop_last' as True in DataLoader in model.py. i.e., train_loader = DataLoader(dataset=train_dataset, batch_size=batch_size, shuffle=True, drop_last = True) valid_loader = DataLoader(dataset=valid_dataset, batch_size=batch_size, shuffle=False, drop_last =...