summaryrefslogtreecommitdiff
path: root/models/auto_encoder.py
AgeCommit message (Collapse)Author
2020-12-23Modify activation functions after conv or trans-conv in auto-encoderJordan Gong
1. Make activation functions be inplace ops 2. Change Leaky ReLU to ReLU in decoder
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