Docker environment for https://github.com/tyliupku/wiki2bio
docker run -it --runtime=nvidia --device /dev/nvidia0:/dev/nvidia0 --device /dev/nvidiactl:/dev/nvidiactl --device /dev/nvidia-uvm:/dev/nvidia-uvm --device /dev/nvidia-modeset:/dev/nvidia-modeset --device /dev/nvidia-uvm-tools:/dev/nvidia-uvm-tools zzj0402/wiki2bio-environment bash
In ~/wiki2bio run python preprocess.py
python Main.py | tee /root/log/training.log
Move files from loads directory to result timestamp home directory:
mv ~/wiki2bio/results/res/1580331814053/loads/69/* ../../
python test.py | tee /root/log/test.log
Content type
Image
Digest
Size
2.9 GB
Last updated
over 6 years ago
docker pull zzj0402/wiki2bio-environment