SampleFactory APPO baseline for MineRL Treechop environment (currently works poorly, the main idea is to test MineRL FPS)
Just install all dependencies using:
pip install -r docker/requirements.txt
Just run train.py
with config_path:
python main.py --config_path treechop_baseline.yaml
We use crafting to automate our experiments.
You can find an example of running such a pipeline in run.yaml
file.
You need to have installed Docker, Nvidia drivers, and crafting package.
The crafting package is available in PyPI:
pip install crafting
To build the image run the command below in docker
folder:
sh build.sh
To run an experiment specify target command in command
field in run.yaml
file and call crafting:
crafting run.yaml
Example of run.yaml
file ():
container:
image: "treechop-appo-baseline:latest"
command: 'python main.py --config_path treechop_baseline.yaml'
tty: True
environment:
- "WANDB_API_KEY=<YOUR API KEY>"
- "OMP_NUM_THREADS=1"
- "MKL_NUM_THREADS=1"
- "NVIDIA_VISIBLE_DEVICES=0"
code:
folder: "."
host_config:
runtime: nvidia
shm_size: 4g
mem_limit: 32g
Please specify your <WANDB_API_KEY> if you want to save logs in wandb cloud or turn off wandb in the training config.