我正在尝试运行urw7rs/spiralpp,这是一个深入学习的项目
首先,我通过以下命令安装了他们的docker映像:
docker run -it -p 8888:8888 urw7rs/spiralpp:latest /bin/bash
下面是Docker映像内部的ls
输出:
(spiralpp) root@8c0404e74c57:/src/spiralpp# ls
CODE_OF_CONDUCT.md
LICENSE
dataset.lock
libtorchbeast.egg-info
pyproject.toml
setup.py
third_party
CelebAMask-HQ
README.md
demo.ipynb
nest
requirements.txt
spiral-envs
torchbeast
Dockerfile
build
libtorchbeast
plot.ipynb
scripts
tests
然后在运行此命令后:
python -m torchbeast.polybeast \
--dataset celeba-hq \
--env_type libmypaint \
--canvas_width 64 \
--use_pressure \
--use_tca \
--num_actors 64 \
--total_steps 30000000 \
--policy_learning_rate 0.0004 \
--entropy_cost 0.01 \
--batch_size 64 \
--episode_length 40 \
--xpid example
我得到了以下错误:
Traceback (most recent call last):
File "/root/miniconda3/envs/spiralpp/lib/python3.7/runpy.py", line 193, in _run_module_as_main
"__main__", mod_spec)
File "/root/miniconda3/envs/spiralpp/lib/python3.7/runpy.py", line 85, in _run_code
exec(code, run_globals)
File "/src/spiralpp/torchbeast/polybeast.py", line 59, in <module>
main()
File "/src/spiralpp/torchbeast/polybeast.py", line 52, in main
run_learner(flags)
File "/src/spiralpp/torchbeast/polybeast.py", line 31, in run_learner
polybeast_learner.main(flags)
File "/src/spiralpp/torchbeast/polybeast_learner.py", line 865, in main
train(flags)
File "/src/spiralpp/torchbeast/polybeast_learner.py", line 585, in train
dataset = utils.create_dataset(flags.dataset, grayscale)
File "/src/spiralpp/torchbeast/utils.py", line 88, in create_dataset
dataset = CelebAHQ(root="./", split="train", transform=tsfm, download=True)
File "/src/spiralpp/torchbeast/core/datasets.py", line 53, in __init__
self.download()
File "/src/spiralpp/torchbeast/core/datasets.py", line 122, in download
os.path.join(self.root, self.base_folder, "CelebAMask-HQ.zip"), "r"
File "/root/miniconda3/envs/spiralpp/lib/python3.7/zipfile.py", line 1258, in __init__
self._RealGetContents()
File "/root/miniconda3/envs/spiralpp/lib/python3.7/zipfile.py", line 1325, in _RealGetContents
raise BadZipFile("File is not a zip file")
zipfile.BadZipFile: File is not a zip file
错误来自第122行/src/spiralpp/torchbeast/core/datasets.py:
with zipfile.ZipFile(
os.path.join(self.root, self.base_folder, "CelebAMask-HQ.zip"), "r"
) as f:
f.extractall(os.path.join(self.root, self.base_folder))
下面是来自CelebAMask-HQ
目录内的ls
的输出:
(spiralpp) root@8c0404e74c57:/src/spiralpp/CelebAMask-HQ# ls
CelebAMask-HQ.zip
list_eval_partition.txt
目前没有回答
相关问题 更多 >
编程相关推荐