Closed esquires closed 5 years ago
You should be good with pytorch version 0.4.1
You can check the requirements.txt for all recommended package versions. If I get some time I'll look into what's changed and see if pytorch 1.x
support isn't too hard.
Thanks! That seems to have fixed it. Feel free to re-open if you want to investigate pytorch 1.x
but this resolves what I was having difficulty with.
When I run the training scenario from the README:
I get the following output:
After setting
torch.autograd.set_detect_anomaly(True)
in the initializer of theQLearner
object I get the following:I then broke-up the line so that the
gather
andsqueeze
functions are on separate lines and the code complains about thegather
line.I am on
33070bc
forpymarl
,cdfff9b
forsmac
, and version1.1.0
forpytorch
.Should I be running a different command than what is in the
README
?Thanks.