├── .gitmodules ├── LICENSE ├── README.md ├── agents ├── __init__.py ├── a2c_base.py ├── base_agent.py ├── curl_dqn_com.py ├── curl_sac.py ├── curl_sacfd.py ├── dqn_agent_com.py ├── dqn_agent_com_drq.py ├── dqn_base.py ├── sac.py ├── sac_aux.py ├── sac_drq.py ├── sacfd.py └── sacfd_drq.py ├── img ├── envs.png └── envs_corl.png ├── networks ├── __init__.py ├── curl_sac_net.py ├── dqn_net.py ├── equivariant_dqn_net.py ├── equivariant_sac_net.py └── sac_net.py ├── requirements.txt ├── scripts ├── __init__.py ├── main.py ├── outputs │ └── __init__.py ├── plot_multiple.py └── test.py ├── storage ├── __init__.py ├── aug_buffer.py ├── buffer.py ├── per_aug_buffer.py ├── per_buffer.py └── segment_tree.py └── utils ├── __init__.py ├── create_agent.py ├── env_wrapper.py ├── logger.py ├── parameters.py ├── schedules.py ├── torch_utils.py └── transformations.py /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/README.md -------------------------------------------------------------------------------- /agents/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /agents/a2c_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/a2c_base.py -------------------------------------------------------------------------------- /agents/base_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/base_agent.py -------------------------------------------------------------------------------- /agents/curl_dqn_com.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/curl_dqn_com.py -------------------------------------------------------------------------------- /agents/curl_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/curl_sac.py -------------------------------------------------------------------------------- /agents/curl_sacfd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/curl_sacfd.py -------------------------------------------------------------------------------- /agents/dqn_agent_com.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/dqn_agent_com.py -------------------------------------------------------------------------------- /agents/dqn_agent_com_drq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/dqn_agent_com_drq.py -------------------------------------------------------------------------------- /agents/dqn_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/dqn_base.py -------------------------------------------------------------------------------- /agents/sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/sac.py -------------------------------------------------------------------------------- /agents/sac_aux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/sac_aux.py -------------------------------------------------------------------------------- /agents/sac_drq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/sac_drq.py -------------------------------------------------------------------------------- /agents/sacfd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/sacfd.py -------------------------------------------------------------------------------- /agents/sacfd_drq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/agents/sacfd_drq.py -------------------------------------------------------------------------------- /img/envs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/img/envs.png -------------------------------------------------------------------------------- /img/envs_corl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/img/envs_corl.png -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/curl_sac_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/networks/curl_sac_net.py -------------------------------------------------------------------------------- /networks/dqn_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/networks/dqn_net.py -------------------------------------------------------------------------------- /networks/equivariant_dqn_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/networks/equivariant_dqn_net.py -------------------------------------------------------------------------------- /networks/equivariant_sac_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/networks/equivariant_sac_net.py -------------------------------------------------------------------------------- /networks/sac_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/networks/sac_net.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/scripts/main.py -------------------------------------------------------------------------------- /scripts/outputs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/plot_multiple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/scripts/plot_multiple.py -------------------------------------------------------------------------------- /scripts/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/scripts/test.py -------------------------------------------------------------------------------- /storage/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /storage/aug_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/storage/aug_buffer.py -------------------------------------------------------------------------------- /storage/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/storage/buffer.py -------------------------------------------------------------------------------- /storage/per_aug_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/storage/per_aug_buffer.py -------------------------------------------------------------------------------- /storage/per_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/storage/per_buffer.py -------------------------------------------------------------------------------- /storage/segment_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/storage/segment_tree.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/create_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/utils/create_agent.py -------------------------------------------------------------------------------- /utils/env_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/utils/env_wrapper.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/utils/parameters.py -------------------------------------------------------------------------------- /utils/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/utils/schedules.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pointW/equi_rl/HEAD/utils/transformations.py --------------------------------------------------------------------------------