lstm_pm_pytorch
lstm_pm_pytorch copied to clipboard
train on penn dataset
thanks for your excellent code, what should I do if i want to train on penn dataset. the dimensions of the value returned in penn_data.py is different . the (h,w) is image height and width but not 368 and the returned shape is 5D in dataloader.