About the encoders, decoders and AII generators

[lyx0208]

According to the pseudo code "Algorithm 1" in the supp.pdf file, there are 2 pretrained encoders, 3 decoders and 2 AII generators.

For the encoders and decoders, if I only use 1 module and use it several times during the training, I have this error:
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation
It's OK to copy a encoder as it's pretrained and I do not need to optimize it during training. But for the decoder, I think its not proper to copy it 3 times, as according to my understanding of the algorithm, the three parts should use decoder with the same parameter.

For the AII generator, I think I just need to have two generators with the same structure, just as the cycle gan has two generators. And I can use the Lcyc in line 41 to optimize these two generators together. Is this understanding right?

[lyx0208]

solved