Open Roller44 opened 1 year ago
Primarily: 1. add default values for "data-folder" and "dataset". 2. correct the error of not using CUDA when it is available. Secondary: correct the wrong dimension of transform for 'mnist', 'fashion-mnist', 'cifar10'.
Primarily: 1. add default values for "data-folder" and "dataset". 2. correct the error of not using CUDA when it is available. Secondary: correct the wrong dimension of transform for 'mnist', 'fashion-mnist', 'cifar10'.