Closed LyndonFan closed 2 years ago
Fixed a couple things for exp 6.2 and allowed VAE to take in CIFAR and other types of layers. Also, setting num_workers=2 in dataloader seemed to not work -- it gave an error of functions can't be pickled. I've deleted that part for now.
Fixed a couple things for exp 6.2 and allowed VAE to take in CIFAR and other types of layers. Also, setting num_workers=2 in dataloader seemed to not work -- it gave an error of functions can't be pickled. I've deleted that part for now.