The reference baseline of final exam for XMU machine learning course

The baseline is a reference method for the final exam of machine learning course.

Requirements

we use /python3.7 /torch 1.4.0+cpu /torchvision 0.5.0+cpu for training and evaluation. You can install the pytorch1.4.0 by using this.

conda install pytorch==1.4.0 torchvision==0.5.0 cpuonly -c pytorch

By the way, you can also use the pytorch with cuda to train this baseline.

You need to create the ./data/ folder and put the ./mini_nico/train and ./mini_nico/test in Mini-NICO dataset to the ./data/ directory like