Giter VIP home page Giter VIP logo

danhrmti / ecrl Goto Github PK

View Code? Open in Web Editor NEW
13.0 2.0 1.0 14.75 MB

Official PyTorch implementation of "Entity-Centric Reinforcement Learning for Object Manipulation from Pixels", Haramati et al., ICLR 2024

Home Page: https://sites.google.com/view/entity-centric-rl

License: MIT License

Batchfile 0.07% Python 99.93%
object-centric-learning reinforcement-learning robotic-manipulation visual-reinforcement-learning

ecrl's People

Contributors

danhrmti avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

Forkers

lc-dev

ecrl's Issues

Results expectation

Hello again,

I have a brief question about the results that I am supposed to get on wandb as well as from running main.py and eval_policy.py.

I'm attempting to train an ordered push policy. I run main.py -c config/ordered_push. In the respective Config.yaml file I specify latentRepPath: 'latent_rep_chkpts/dlp_push_5C' for pre-trained DLP and train with numObjects = 2. I run policy_eval.py config/ordered_push with the respective checkpoint found in model_chkpts and the same number of objects.

I am unsure of what I am supposed to expect in the results. My issues are as follows:

  1. There are several .gif files that I obtained in wanbd: Eval Goal Image, Goal Image - Frontview, Goal Image - Sideview, Episode Video - Frontview, Episode Video - Sideview, and Eval Episode Video. I'm getting videos for both main.py and policy_eval.py. I'm not sure about what these videos represent and at which point in the training they are recorded.

  2. I notice that the code runs after 800 episodes. I'm probably misunderstanding the code a little bit, but I'm currently expecting that the code should stop as it is specified in the config file.

  3. I would be curious to see what are my expected graphs for obs_rms_std and obs_rms_mean should look like. I see the error decrease, but only from 0.68 to 0.655 and -0.03 to -0.04 respectively (which tells me that I'm probably not learning much). Would you mind telling me what results you got on that end?

Thank you very much for your help!

Hanna

p.s. Would it be possible to email you for further personal issues I have?

Dependency issues

Hi,

While trying to run your code, I ran into several dependency issues.

In the prerequisites section, you mention having torch 2.0.1 and stable-baseline3 1.5.0. However, your requirements.txt has torch 2.1.2. Furthermore, I can't install gym 0.21.0 without downgrading to python 3.7. I have more similar issues that require newer versions of stable-baseline3.

Also, I have permission denied when trying to run python main.py -c config/push_t (for wandb).

Would you mind helping me to get started running your code in conda?

Thanks!

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.