mila-iqia / atari-representation-learning

Code for "Unsupervised State Representation Learning in Atari"
https://arxiv.org/abs/1906.08226
MIT License
235 stars 50 forks source link

Evaluate "transporter" network #51

Open DuaneNielsen opened 4 years ago

DuaneNielsen commented 4 years ago

Awesome work guys. Can't wait to take this for a spin.

Recently Mnih's group released a paper on representation learning using Atari.

Unsupervised Learning of Object Keypoints for Perception and Control

https://arxiv.org/abs/1906.11883

Are you guys intending to benchmark their transporter network?

I'd be interested to know your thoughts. (I also have a working pytorch implementation of the transporter network).

ankeshanand commented 4 years ago

Thanks @DuaneNielsen!

We've read the Keypoints paper, it's a really cool work! I don't know if we will be able to get to benchmarking it soon enough though.

If you are planning to benchmark it, that would be great and we will be glad to help you. Feel free to email us / raise an issue if you have any questions.

DuaneNielsen commented 4 years ago

Thanks, I'll pull down your code and see if I can get it running here. Will let you know.