summaryrefslogtreecommitdiff
path: root/models/auto_encoder.py
AgeCommit message (Collapse)Author
2020-12-23Refactor and refine auto-encoderJordan Gong
1. Wrap Conv2d 3x3-padding-1 to VGGConv2d 2. Wrap ConvTranspose2d 4x4-stride-4-padding-1 to DCGANConvTranspose2d 3. Turn off bias in conv since the employment of batch normalization
2020-12-23Reshape feature before decodeJordan Gong
2020-12-23Split modules to different filesJordan Gong