hello. I’m in trouble using TransformerEncoderLayer.
my torch version is 1.8.1+cu102
when I use batch_first, It shows error.
torch.nn.TransformerEncoderLayer(d_model=time_step, nhead=4, dropout=0.2, batch_first=True)
TypeError: init() got an unexpected keyword argument ‘batch_first’