add AMP to trainer.py - Githubissues

duanzhiihao / lossy-vae

Authors' PyTorch implementation of lossy image compression methods based on hierarchical VAEs

54 stars 6 forks source link

Closed duanzhiihao closed 1 year ago