PyTorch implementation of Advantage async actor-critic Algorithms (A3C) in PyTorch

Overview

Advantage async actor-critic Algorithms (A3C) in PyTorch

@inproceedings{mnih2016asynchronous,
  title={Asynchronous methods for deep reinforcement learning},
  author={Mnih, Volodymyr and Badia, Adria Puigdomenech and Mirza, Mehdi and Graves, Alex and Lillicrap, Timothy P and Harley, Tim and Silver, David and Kavukcuoglu, Koray},
  booktitle={International Conference on Machine Learning},
  year={2016}}

This repository contains an implementation of Adavantage async Actor-Critic (A3C) in PyTorch based on the original paper by the authors and the PyTorch implementation by Ilya Kostrikov.

A3C is the state-of-art Deep Reinforcement Learning method.

Dependencies

  • Python 2.7
  • PyTorch
  • gym (OpenAI)
  • universe (OpenAI)
  • opencv (for env state processing)
  • visdom (for visualization)

Training

./train_lstm.sh

Test wigh trained weight after 169000 updates for PongDeterminisitc-v3.

./test_lstm.sh 169000

A test result video is available.

Check the loss curves of all threads in http://localhost:8097

loss_png

References

You might also like...
Implement A3C for Mujoco gym envs
Implement A3C for Mujoco gym envs

pytorch-a3c-mujoco Disclaimer: my implementation right now is unstable (you ca refer to the learning curve below), I'm not sure if it's my problems. A

Scripts of Machine Learning Algorithms from Scratch. Implementations of machine learning models and algorithms using nothing but NumPy with a focus on accessibility. Aims to cover everything from basic to advance.
Scripts of Machine Learning Algorithms from Scratch. Implementations of machine learning models and algorithms using nothing but NumPy with a focus on accessibility. Aims to cover everything from basic to advance.

Algo-ScriptML Python implementations of some of the fundamental Machine Learning models and algorithms from scratch. The goal of this project is not t

Rate-limit-semaphore - Semaphore implementation with rate limit restriction for async-style (any core)
Rate-limit-semaphore - Semaphore implementation with rate limit restriction for async-style (any core)

Rate Limit Semaphore Rate limit semaphore for async-style (any core) There are t

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

Faster R-CNN and Mask R-CNN in PyTorch 1.0 maskrcnn-benchmark has been deprecated. Please see detectron2, which includes implementations for all model

PyTorch implementation of Federated Learning with Non-IID Data, and federated learning algorithms, including FedAvg, FedProx.
PyTorch implementation of Federated Learning with Non-IID Data, and federated learning algorithms, including FedAvg, FedProx.

Federated Learning with Non-IID Data This is an implementation of the following paper: Yue Zhao, Meng Li, Liangzhen Lai, Naveen Suda, Damon Civin, Vik

Pytorch implementation of various High Dynamic Range (HDR) Imaging algorithms

Deep High Dynamic Range Imaging Benchmark This repository is the pytorch impleme

Robust, modular and efficient implementation of advanced Hamiltonian Monte Carlo algorithms

AdvancedHMC.jl AdvancedHMC.jl provides a robust, modular and efficient implementation of advanced HMC algorithms. An illustrative example for Advanced

Implementation of fast algorithms for Maximum Spanning Tree (MST) parsing that includes fast ArcMax+Reweighting+Tarjan algorithm for single-root dependency parsing.

Fast MST Algorithm Implementation of fast algorithms for (Maximum Spanning Tree) MST parsing that includes fast ArcMax+Reweighting+Tarjan algorithm fo

Implementation of algorithms for continuous control (DDPG and NAF).

DEPRECATION This repository is deprecated and is no longer maintaned. Please see a more recent implementation of RL for continuous control at jax-sac.

Comments
  • modprobe: ERROR: could not insert 'fglrx': Function not implemented

    modprobe: ERROR: could not insert 'fglrx': Function not implemented

    After pip install gym[atari], pip install scipy and change PongDeterministic-v3 to PongDeterministic-v4, A3C-PyTorch works well on my computer. But when I run ./train_lstm.sh on the server through ssh -X, the following errors occur. I believe it is due to the incorrect configuration of the server.

    (A3C-Torch) jack@gpu02:/SSD2/jack/Applications/A3C-Torch$ ./train_lstm.sh 
    It's Alive!
    [2017-08-03 14:10:21,731] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:22,112] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:22,453] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:22,735] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:22,977] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:23,225] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:23,467] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:23,714] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:23,956] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:24,200] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:24,442] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:24,686] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:24,928] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:25,170] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:25,414] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:25,657] Making new env: PongDeterministic-v4
    [2017-08-03 14:10:25,899] Making new env: PongDeterministic-v4
    ======training=====
    [2017-08-03 14:10:26,226] Making new env: PongDeterministic-v4
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    modprobe: ERROR: ../libkmod/libkmod-module.c:809 kmod_module_insert_module() could not find module by name='fglrx'
    modprobe: ERROR: could not insert 'fglrx': Function not implemented
    Error: Fail to load fglrx kernel module!
    Error! Fail to load fglrx kernel module! Maybe you can switch to root user to load kernel module directly
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    step:  0
    Intel MKL FATAL ERROR: Error on loading function mkl_blas_avx2_get_kernel_api_version.
    

    My computer is Ubuntu16.04 x64 without cuda. The server is running on Ubuntu 14.04 x64 with cuda 8.0

    Do you have any ideas?

    Thank you!

    opened by Jiankai-Sun 3
Owner
LEI TAI
Autonomous Driving Solutions
LEI TAI
PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

Ilya Kostrikov 3k Dec 31, 2022
Advantage Actor Critic (A2C): jax + flax implementation

Advantage Actor Critic (A2C): jax + flax implementation Current version supports only environments with continious action spaces and was tested on muj

Andrey 3 Jan 23, 2022
Asynchronous Advantage Actor-Critic in PyTorch

Asynchronous Advantage Actor-Critic in PyTorch This is PyTorch implementation of A3C as described in Asynchronous Methods for Deep Reinforcement Learn

Reiji Hatsugai 38 Dec 12, 2022
Softlearning is a reinforcement learning framework for training maximum entropy policies in continuous domains. Includes the official implementation of the Soft Actor-Critic algorithm.

Softlearning Softlearning is a deep reinforcement learning toolbox for training maximum entropy policies in continuous domains. The implementation is

Robotic AI & Learning Lab Berkeley 997 Dec 30, 2022
Using deep actor-critic model to learn best strategies in pair trading

Deep-Reinforcement-Learning-in-Stock-Trading Using deep actor-critic model to learn best strategies in pair trading Abstract Partially observed Markov

null 281 Dec 9, 2022
Multi-task Multi-agent Soft Actor Critic for SMAC

Multi-task Multi-agent Soft Actor Critic for SMAC Overview The CARE formulti-task: Multi-Task Reinforcement Learning with Context-based Representation

RuanJingqing 8 Sep 30, 2022
Independent and minimal implementations of some reinforcement learning algorithms using PyTorch (including PPO, A3C, A2C, ...).

PyTorch RL Minimal Implementations There are implementations of some reinforcement learning algorithms, whose characteristics are as follow: Less pack

Gemini Light 4 Dec 31, 2022
A3C LSTM Atari with Pytorch plus A3G design

NEWLY ADDED A3G A NEW GPU/CPU ARCHITECTURE OF A3C FOR SUBSTANTIALLY ACCELERATED TRAINING!! RL A3C Pytorch NEWLY ADDED A3G!! New implementation of A3C

David Griffis 532 Jan 2, 2023
Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...

Automatic, Readable, Reusable, Extendable Machin is a reinforcement library designed for pytorch. Build status Platform Status Linux Windows Supported

Iffi 348 Dec 24, 2022
Implementation of Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learning

advantage-weighted-regression Implementation of Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learning, by Peng et al. (

Omar D. Domingues 1 Dec 2, 2021