![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
vq-vae-2填坑
忘了有多久·
这个作者很懒,什么都没留下…
展开
-
vq-vae-2代码之cantpickle Environment objects
运行vq-vae-2-pytorch项目,第二阶段: python extract_code.py --ckpt checkpoint/vqvae_560.pt --name lmdbdata dataset 报错TypeError: can’t pickle Environment objects 源代码段为: if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('--batch原创 2020-06-29 15:57:26 · 666 阅读 · 0 评论 -
vq-vae-2代码填坑之CUDA out of memory和element 0 of tensors does not require grad and...
运行vq-vae-2-pytorch项目,第三阶段: python train_pixelsnail.py lmdbdata 报错RuntimeError: CUDA out of memory. Tried to allocate 1.50 GiB… 在out前加入with torch.no_grad(): 触发另外一个错误: RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn 源代码段为原创 2020-06-29 15:42:53 · 540 阅读 · 0 评论