meshed-memory-transformer icon indicating copy to clipboard operation
meshed-memory-transformer copied to clipboard

Published 20 hours ago • verified publisher icon aimagelab

register_state or register_buffer ?

Open xuewyang opened this issue 3 years ago • 1 comments

Hi, Thank you for open-sourcing your codes. I really enjoyed reading your paper. I am having a problem when try to understand:

  1. here. How does register_state work here? Does it have any difference with register_buffer?
  2. Does the enc_output and mask_enc defined in register_state have something to do with the output here?

Thank you for you time.

xuewyang avatar Dec 09 '20 20:12 xuewyang

How long does it take to train the network?

Jennifer-6 avatar Sep 15 '21 11:09 Jennifer-6