jsikyoon/dreamer-torch jsikyoon/dreamer-torch

  • Stars
    star
    111
  • Rank 314,510 (Top 7 %)
  • Language
    Python
  • License
    MIT License
  • Created over 3 years ago
  • Updated almost 3 years ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
jsikyoon/dreamer-torch
github

jsikyoon

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Pytorch version of Dreamer, which follows the original TF v2 codes.

dreamer-torch

Pytorch version of Dreamer, which follows the original TF v2 codes (https://github.com/danijar/dreamerv2/tree/faf9e4c606e735f32c8b971a3092877265e49cc2).

Due to limitation of resource, we tested it with the official TF version2 codes for 10 tasks (5 DMC / 5 Atari tasks) as below.

alt text

As we can see, for almost tasks, it shows similar performance with Dreamer when comparing the reported performance on paper and from running codes.

For freeway, I found that it is slower than Dreamer, maybe I think it is because of random seed. I will find the reason more.

The below logs are from Tensorboard, you also can see through Tensorboard logs.

  • DMC walker walk task

alt text

  • Atari bank heist task

alt text

  • atari freeway task

alt text

How to use

For required packages to run this, you can find from requirements.txt.

The command to run is exactly same to the official codes, you can use it.

Contact

Any feedback are welcome! Please open an issue on this repository or send email to Jaesik Yoon ([email protected]).

More Repositories

1

adv_attack_capsnet

Tensorflow Implementation of Adversarial Attack to Capsule Networks
Python
174
star
2

Interaction-networks_tensorflow

Tensorflow Implementation of Interaction Networks for Learning about Objects, Relations and Physics
Python
158
star
3

visual-interaction-networks_tensorflow

Tensorflow Implementation of Visual Interaction Networks
Python
132
star
4

pathnet

Tensorflow Implementation of PathNet: Evolution Channels Gradient Descent in Super Neural Networks
Python
102
star
5

SequentialData-GAN

Tensorflow Implementation of GAN modeling for sequential data
Python
69
star
6

bmaml

This repository contains implementations of the paper, Bayesian Model-Agnostic Meta-Learning.
Python
56
star
7

programmable-agents_tensorflow

Tensorflow Implementation of Programmable Agents
Python
36
star
8

a3c-distributed_tensorflow

Distributed Tensorflow Implementation of Asynchronous Methods for Deep Reinforcement Learning
Python
31
star
9

bmaml_rl

This repository contains implementations of the paper, Bayesian Model-Agnostic Meta-Learning.
Python
19
star
10

V-MPO_torch

V-MPO torch version with DMLab30 and GTrXL
Python
12
star
11

Asynchronous-DDPG_distributed_tensorflow

Distributed Tensorflow Implementation of Asynchronous DDPG
Python
12
star
12

tree-structured-group-lasso

Python Implementation of Proximal Methods for Hierarchical Sparse Coding
Python
10
star
13

nips17_adv_attack

Tensorflow Implementation of adversarial learning based adversarial example generator
Python
9
star
14

OCRL

Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to RL tasks.
Python
9
star
15

ddocker

Python
8
star
16

svpg_tensorflow

Tensorflow Implementation of Stein Variational Policy Gradients
Python
8
star
17

ASNP-RMR

This is an official Tensorflow implementation of the ASNP-RMR.
Python
7
star