What shoult I do if I want to use this algorithm in orther models and dataset?

[chl2]

I am re-implementing your code in other dataset like Imagenet64, other models like resnet50. How you train the encoder_weight_file and decoder_weight_file ? Can I use it directly?
Thank You.

[chunchentu]

Hi,

Thanks for being interested in our work. The information of the codec is defined in setup_codec. You can then train encoder/decoder based on the model definition.