This project is an implementation of auto-encoder with MNIST dataset and pytorch = 1.4.0
Learn tensorboard by the way
bash ./script/prepare.sh
bash ./script/train.sh ${model_name}
${model_name}
can be one of conv/linear/vae
...
tensorboard --logdir=runs --bind_all
The output shape of ConvTranspose2d
can be computed by formula:
Hung-yi Lee MachineLearning Spring2020 Course (https://www.bilibili.com/video/BV1JE411g7XF?p=59)
https://github.com/L1aoXingyu/pytorch-beginner (Conv Model from here, But train code with this repositories have a bug in transform operation, detail in this pr page)