- 博客(2)
- 收藏
- 关注
RSS订阅原创 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
673
原创 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
555
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人