- 博客(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
750
原创 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
645
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人