python - 3.7.4
numpy - 1.19.2
cudatoolkit - 10.2.89
cudnn - 7.6.5
pytorch - 1.6.0
Please run each .py file separately in the project path.
Larger datasets like MNIST are not contained in the "datasets" folder. Please redirect routes inside the codes.