Encoder-decoder architecture

[astrogilda]

Hey Paul,

Thanks for this repository! Could you help me understand how one would modify the code to function in an encoder decoder capacity, as referenced in the paper?

[ptuls]

Hey @astrogilda,

You'll need to create a new decoder function, similar to the encoder function

def decoder(self, data, reuse, is_training)

You'll have to implement a new loss function as per the paper for self-supervision

in Section 4.

Then, in here, add a decoder after the encoder to train I think. You might want to remove the classification part too.

At least that's the rough idea I have. Hopefully that helps.