TypeError: forward() got an unexpected keyword argument 'src_key_padding_mask'

[luozhiping]

param error use imitate_episodes.py to train model.

TypeError: forward() got an unexpected keyword argument 'src_key_padding_mask'
TypeError: forward() got an unexpected keyword argument 'pos'

at detr_vae.py line 116:          encoder_output = self.encoder(encoder_input, pos=pos_embed, src_key_padding_mask=is_pad)

where transformer.py forward function param is     
def forward(self, src, mask, query_embed, pos_embed, latent_input=None, proprio_input=None, additional_pos_embed=None)

Are there any code updated haven't pushed?

[luozhiping]

change input param to self.encoder(encoder_input, pos_embed=pos_embed, mask=is_pad) and then got this error:
: forward() missing 1 required positional argument: 'query_embed'

[luozhiping]

is there any wrong here? actually it should be encoder = build_encoder(args)

[apirrone]

Thanks @luozhiping

Did just that and it seems to be training now !

in detr/models/detr_vae.py at line 285 change build_transformer to build_encoder

[Ke-Wang1017]

Thanks @luozhiping

Did just that and it seems to be training now !

in detr/models/detr_vae.py at line 285 change build_transformer to build_encoder

This works for me as well!