Rohan Potdar
Rohan Potdar
You might also want to try: https://github.com/NVlabs/cule/issues/34#issuecomment-1166889998
Hi there-I got it to work on my laptop (GTX 1650 Ti, CUDA 11.3, PyTorch 1.11.0) by fixing the following lines in `setup.py`: ``` codes = [arch[-2:] for arch in...
Not yet-These fixes are for `ray==2.0.0`, which hasn't been released on PyPI yet. I'll un-draft this PR when it's released next week.
I can add you to my fork, but do you want to add my repo as a remote and try `git checkout -t Rohan138/dreamer`?
Just sent you a contributor invite; maybe `https://github.com/Rohan138/mbrl-lib.git` would've worked better than `git@github`?
On the non-Dreamer fixes: - I renamed `pyproyect.toml` to `pyproject.toml` - I removed the lines pinning `python==3.7` on all commits - There's no black version 21.4b2 on pypi, so I...
Hi! So I tried running it, but despite the losses dropping, it doesn't seem to learn right now. [Here](https://drive.google.com/file/d/1AgeveNg2iDIBop26PJcFMFCOVvOGu_Ed/view?usp=sharing) are the results; I'm planning to look through the `agent.train()` implementation...
Sorry for the delay-I'll try to address all of the comments across the next day or so. I moved the non-Dreamer fixes to #161. The `compute_return` is taken from [this](https://github.com/juliusfrost/dreamer-pytorch/blob/master/dreamer/algos/dreamer_algo.py#L331)...
Hi! Just wanted to check in on progress on this; I'd be interested in contributing as well.
@luisenp Just opened a draft PR #151 ; mainly need to debug and clean up the dreamer loss calculation, then test it.