pytorch-vqvae icon indicating copy to clipboard operation
pytorch-vqvae copied to clipboard
verified publisher icon ritheshkumar95

Enable run main without argparse

Open Roller44 opened this issue 2 years ago • 0 comments

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'.

Roller44 avatar Jul 11 '23 02:07 Roller44