Resnet 18 at first tr
example usage
optional arguments:
-h, --help show this help message and exit
-dir DIR path
-device DEVICE device
-savepath SAVEPATH savepath
-num_epochs NUM_EPOCHS
number of epochs
-batch_size BATCH_SIZE
batch size