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', type=int, default=16)
    parser.add_argument('--epoch', type=int, default=420)
    parser.add_argument('--hier', type=str, default='top')
    parser.add_argument('--lr', type=float, default=3e-4)
    parser.add_argument('--channel', type=int, default=256)
    parser.add_argument('--n_res_block', type=int, default=4)
    parser.