summaryrefslogtreecommitdiff
path: root/models/auto_encoder.py
AgeCommit message (Collapse)Author
2020-12-27Fix inconsistency and API deprecation issues in decoderJordan Gong
1. Add default output channels of decoder 2. Replace deprecated torch.nn.functional.sigmoid with torch.sigmoid
2020-12-27Refine auto-encoderJordan Gong
1. Wrap fully connected layers 2. Introduce hyperparameter tuning in constructor
2020-12-24Optimize importsJordan Gong
2020-12-24Change the usage of layers and reorganize relations of layersJordan Gong
1. Add batch normalization and activation to layers 2. VGGConv2d and FocalConv2d inherits to BasicConv2d; DCGANConvTranspose2d inherits to BasicConvTranspose2d
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