torch >= 1.7.0
yaml
omegaconf
visdom
torchmetrics
Pillow 8.2.0
PrettyTable
you can directly run the code with the corrosponding dataset name, for example, the simulation can start with
python mnist.py
- The code is provided by Hui Zeng from NUDT. If you have any question, please contact [email protected].