README
The code is based on the ILswiss.
To run the code, use
python run_experiment.py --nosrun -e <your YAML file> -g <gpu id>
Generally, run_experiment.py loads the YAML file, creating multiple processes, each of which runs the script assigned in the YAML file.
The script of COIL is run_scripts/coil_script.py. Dataset settings are in demos_listing.yaml. The core algorithm is in rlkit/torch/coil/coil.py. New algorithms should also be put under similar directories. A trajectory replay buffer and the trajectory picking algorithm is in rlkit/data_management/episodic_replay_buffer_coil.py.