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 Installation 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. Prepare Datasets 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
Read more