CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning.
MishaLaskin/curl is ranked 16,186 with 607 stars.
Same-category comparison
MishaLaskin/curl is ranked 16,186 among these 8 peers; theAIGuysCode/yolov4-custom-functions is ranked 16,182.
Same-category comparison| Name | Rank | Metrics |
|---|
| benedekrozemberczki/GraphWaveletNeuralNetwork | 16,172 | - Language:
- Python
- Stars:
- 608
- Forks:
- 117
- Description:
- A PyTorch implementation of "Graph Wavelet Neural Network" (ICLR 2019).
|
|---|
| google-research-datasets/dstc8-schema-guided-dialogue | 16,175 | - Language:
- Python
- Stars:
- 608
- Forks:
- 136
- Description:
- The Schema-Guided Dialogue Dataset.
|
|---|
| stefanopini/simple-HRNet | 16,180 | - Language:
- Python
- Stars:
- 608
- Forks:
- 111
- Description:
- Multi-person Human Pose Estimation with HRNet in Pytorch.
|
|---|
| theAIGuysCode/yolov4-custom-functions | 16,182 | - Language:
- Python
- Stars:
- 608
- Forks:
- 360
- Description:
- A Wide Range of Custom Functions for YOLOv4, YOLOv4-tiny, YOLOv3, and YOLOv3-tiny Implemented in TensorFlow, TFLite, and TensorRT.
|
|---|
| wzdnzd/harvester | 16,193 | - Language:
- Python
- Stars:
- 607
- Forks:
- 136
- Description:
- Intelligent data acquisition framework for GitHub and web sources.
|
|---|
| EdGENetworks/attention-networks-for-classification | 16,196 | - Language:
- Jupyter Notebook
- Stars:
- 606
- Forks:
- 130
- Description:
- Hierarchical Attention Networks for Document Classification in PyTorch.
|
|---|
| levyflux/AViD | 16,205 | - Language:
- Python
- Stars:
- 606
- Forks:
- 95
- Description:
- Framework that enables fine-tuning of vision-language grounding models on custom datasets.
|
|---|
| microsoft/CvT | 16,207 | - Language:
- Python
- Stars:
- 606
- Forks:
- 130
- Description:
- This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.
|
|---|
Key parameters and specifications
Key parameters and specifications| Parameter | Value |
|---|
| Language | Python |
|---|
| Category | Model Development |
|---|
| Subcategory | Reinforcement Learning |
|---|
| Keywords | Computer vision, RL |
|---|
| Stars | 607 |
|---|
| Forks | 92 |
|---|
| Contributors | 4 |
|---|
| Created date | 2020-02-09 |
|---|
| Updated date | 2026-09-22 |
|---|
| Stars in 1 day | 0 |
|---|
| Stars in 7 days | 0 |
|---|
| Rank | 16,186 |
|---|
| Previous rank | 15,973 |
|---|
| Rank change | -213 |
|---|
| 7-day stars rank | 15,389 |
|---|
| Previous 7-day stars rank | 15,397 |
|---|
| 7-day stars rank change | 8 |
|---|
| Total stars rank | 16,640 |
|---|
| Previous total stars rank | 16,613 |
|---|
| Total stars rank change | -27 |
|---|
| Country | USA |
|---|
README summary
CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning This repository is the official implementation of CURL for the DeepMind control experiments. Atari experiments were done in a separate codebase available here. Our implementation of SAC is based on SAC+AE by Denis Yarats. Installation All of the dependencies are in the file. They can be installed manually or with the following command: Instructions To train a CURL agent on the task from image-based observations run from the
root of this directory. The file contains the following command, which you can modify to try different environments / hyperparamters. In your console, you should see printouts that look like: For reference, the maximum score for cartpole swing up is around 845 pts, so CURL has converged to the optimal score. This takes about an hour of training depending on your GPU. Log abbreviation mapping: All data related to the run is stored in the specified . To enable model or video
saving, use the or flags. To visualize progress with tensorboard run: and go to in your browser. If you're running headlessly, try port forwarding with ssh. For GPU accelerated rendering, make sure EGL is installed on your machine and set . For environment troubleshooting issues, see the DeepMind control documentation.
Frequently asked questions
How many stars and forks are recorded?
MishaLaskin/curl has 607 stars and 92 forks.
What is the repository rank?
MishaLaskin/curl is ranked 16,186.
What language and category are recorded?
MishaLaskin/curl has 607 stars, uses Python and belongs to Model Development.
Which repository is closest in rank?
MishaLaskin/curl is compared with theAIGuysCode/yolov4-custom-functions, ranked 16,182 with 608 stars.
CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning
This repository is the official implementation of CURL for the DeepMind control experiments. Atari experiments were done in a separate codebase available here. Our implementation of SAC is based on SAC+AE by Denis Yarats.
Installation
All of the dependencies are in the conda_env.yml file. They can be installed manually or with the following command:
conda env create -f conda_env.yml
Instructions
To train a CURL agent on the cartpole swingup task from image-based observations run bash script/run.sh from the root of this directory. The run.sh file contains the following command, which you can modify to try different environments / hyperparamters.
CUDA_VISIBLE_DEVICES=0 python train.py \
--domain_name cartpole \
--task_name swingup \
--encoder_type pixel \
--action_repeat 8 \
--save_tb --pre_transform_image_size 100 --image_size 84 \
--work_dir ./tmp \
--agent curl_sac --frame_stack 3 \
--seed -1 --critic_lr 1e-3 --actor_lr 1e-3 --eval_freq 10000 --batch_size 128 --num_train_steps 1000000
In your console, you should see printouts that look like:
| train | E: 221 | S: 28000 | D: 18.1 s | R: 785.2634 | BR: 3.8815 | A_LOSS: -305.7328 | CR_LOSS: 190.9854 | CU_LOSS: 0.0000
| train | E: 225 | S: 28500 | D: 18.6 s | R: 832.4937 | BR: 3.9644 | A_LOSS: -308.7789 | CR_LOSS: 126.0638 | CU_LOSS: 0.0000
| train | E: 229 | S: 29000 | D: 18.8 s | R: 683.6702 | BR: 3.7384 | A_LOSS: -311.3941 | CR_LOSS: 140.2573 | CU_LOSS: 0.0000
| train | E: 233 | S: 29500 | D: 19.6 s | R: 838.0947 | BR: 3.7254 | A_LOSS: -316.9415 | CR_LOSS: 136.5304 | CU_LOSS: 0.0000
For reference, the maximum score for cartpole swing up is around 845 pts, so CURL has converged to the optimal score. This takes about an hour of training depending on your GPU.
Log abbreviation mapping:
train - training episode
E - total number of episodes
S - total number of environment steps
D - duration in seconds to train 1 episode
R - mean episode reward
BR - average reward of sampled batch
A_LOSS - average loss of actor
CR_LOSS - average loss of critic
CU_LOSS - average loss of the CURL encoder
All data related to the run is stored in the specified working_dir. To enable model or video saving, use the --save_model or --save_video flags. For all available flags, inspect train.py. To visualize progress with tensorboard run:
tensorboard --logdir log --port 6006
and go to localhost:6006 in your browser. If you're running headlessly, try port forwarding with ssh.
For GPU accelerated rendering, make sure EGL is installed on your machine and set export MUJOCO_GL=egl. For environment troubleshooting issues, see the DeepMind control documentation.