search

castorini / castor

PyTorch deep learning models for text processing
http://castor.ai/
Apache License 2.0
178 stars 58 forks source link

Fix HAN for batch_size 1 #161

Closed Ashutosh-Adhikari closed 5 years ago

Ashutosh-Adhikari commented 5 years ago

Current implementation of HAN throws an error due to dimension not specified to squeeze during the forward pass. Specify dimension for the same.