CNN-TE

Running relies on run_main.py. However, edit this file and specify which dataset you want to run by uncommenting the required key (that is identical with the root key in the RunConfig.py file) at the beginning of the file, just after the imports. This will download the dataset specified by the key. It will also generate a best mode .pth file and also a checkpoint model in the same path as the run_main.py.