Reworked at 20221123
You can run train code by train.py
conda env create -f environment.yaml
conda activate ensnet
python .\train.py dataset=mnist # [mnist, fashion_mnist, cifar10] Or you can use --multirun. Check hydra
Pytorch implementation of EnsNet (Unofficial).
Reworked at 20221123
You can run train code by train.py
conda env create -f environment.yaml
conda activate ensnet
python .\train.py dataset=mnist # [mnist, fashion_mnist, cifar10] Or you can use --multirun. Check hydra
Pytorch implementation of EnsNet (Unofficial).