diff --git a/.gitignore b/.gitignore index 1be77204..6dd52f03 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,6 @@ +.vscode/ +.d4rl/ + bin/ logs/ wandb/ diff --git a/README.md b/README.md index c6e2b8b5..e01a5189 100644 --- a/README.md +++ b/README.md @@ -44,11 +44,11 @@ docker build -f azure/Dockerfile . -t diffuser ``` docker run -it --rm --gpus all \ --mount type=bind,source=$PWD,target=/home/code \ + --mount type=bind,source=$PWD/logs,target=/logs/docker \ --mount type=bind,source=$HOME/.d4rl,target=/root/.d4rl \ diffuser \ bash -c \ - "export PYTHONPATH=$PYTHONPATH:/home/code && \ - python /home/code/scripts/train.py --dataset hopper-medium-expert-v2 --logbase logs/docker" + "python /home/code/scripts/train.py --dataset hopper-medium-expert-v2 --logbase logs/docker" ``` diff --git a/docker-interactive.sh b/docker-interactive.sh new file mode 100644 index 00000000..e6898c99 --- /dev/null +++ b/docker-interactive.sh @@ -0,0 +1 @@ +docker run -it --rm --gpus all --mount type=bind,source=$PWD,target=/home/code --mount type=bind,source=$PWD/logs,target=/logs/docker --mount type=bind,source=$PWD/.d4rl,target=/root/.d4rl -p 1222:1222 diffuser \ No newline at end of file