Offline Reinforcement Learning with Implicit Q-Learning

This repository contains the official implementation of Offline Reinforcement Learning with Implicit Q-Learning by Ilya Kostrikov, Ashvin Nair, and Sergey Levine.

If you use this code for your research, please consider citing the paper:

@article{kostrikov2021iql,
    title={Offline Reinforcement Learning with Implicit Q-Learning},
    author={Ilya Kostrikov and Ashvin Nair and Sergey Levine},
    year={2021},
    archivePrefix={arXiv},
    primaryClass={cs.LG}
}

How to run the code

Install dependencies

pip install -r requirements.txt

See instructions for CUDA.

Run training

Locomotion

python train_offline.py --env_name=halfcheetah-medium-expert-v2 --config=configs/mujoco_config.py

AntMaze

python train_offline.py --env_name=antmaze-large-play-v0 --config=configs/antmaze_config.py --eval_episodes=100 --eval_interval=100000

Kitchen and Adroit

python train_offline.py --env_name=pen-human-v0 --config=configs/kitchen_config.py

Misc

The implementation is based on JAXRL.

Offline Reinforcement Learning with Implicit Q-Learning

Related tags

Overview

Offline Reinforcement Learning with Implicit Q-Learning

How to run the code

Install dependencies

Run training

Misc

Owner

Ilya Kostrikov

[ACM MM 2019 Oral] Cycle In Cycle Generative Adversarial Networks for Keypoint-Guided Image Generation

Example-custom-ml-block-keras - Custom Keras ML block example for Edge Impulse

SynNet - synthetic tree generation using neural networks

It is modified Tensorflow 2.x version of Mask R-CNN

The final project of "Applying AI to 2D Medical Imaging Data" of "AI for Healthcare" nanodegree - Udacity.

ReferFormer - Official Implementation of ReferFormer

Ansible Automation Example: JSNAPY PRE/POST Upgrade Validation

Locally Enhanced Self-Attention: Rethinking Self-Attention as Local and Context Terms

EquiBind: Geometric Deep Learning for Drug Binding Structure Prediction

Predictive AI layer for existing databases.

Picasso: A CUDA-based Library for Deep Learning over 3D Meshes

Image Completion with Deep Learning in TensorFlow

WatermarkRemoval-WDNet-WACV2021

Generative Exploration and Exploitation - This is an improved version of GENE.

这是一个利用facenet和retinaface实现人脸识别的库，可以进行在线的人脸识别。

FFTNet vocoder implementation

Simple and Distributed Machine Learning

Self-supervised Augmentation Consistency for Adapting Semantic Segmentation (CVPR 2021)

This is the code repository for the paper A hierarchical semantic segmentation framework for computer-vision-based bridge column damage detection

Edison AT is software Depression Assistant personal.