hkproj/pytorch-transformer

forward method in transformer class

Closed this issue · 1 comments

akkasi commented

Hi,
It seems there is something wrong with transformer class. It does not have a forward method! could you please clarify the reason? Tnx

hkproj commented

Hi! The forward method is not needed, as we apply the encoder and decoder "by hand":

pytorch-transformer/train.py

Line 234 in 1af7967

encoder_output = model.encode(encoder_input, encoder_mask) # (B, seq_len, d_model)