Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

PyTorch 1.8.1 Compatibility with Training and Evaluation #15

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

atagade
Copy link

@atagade atagade commented Aug 25, 2021

Expanding on the PR's raised by #12 and #10, this PR focuses on providing compatibility for PyTorch 1.8.1 for both evaluation and training purposes. With this PR it is possible to train agents using different label mappings and feature maps.

@Throne-myworld
Copy link

Expanding on the PR's raised by #12 and #10, this PR focuses on providing compatibility for PyTorch 1.8.1 for both evaluation and training purposes. With this PR it is possible to train agents using different label mappings and feature maps.
But when I running the code on win10 and using the latest python,pytorch version is 1.8.1,I got this
quesiton:
File "D:\PyCharmproject\Arnold-master\Arnold-master\src\model\dqn\base.py", line 81, in base_forward
output = torch.cat([conv_output] + embeddings, dim=1)
RuntimeError: torch.cat(): Tensors must have same number of dimensions: got 2 and 3
which is quiet confusing for me to solve it

@atagade
Copy link
Author

atagade commented Nov 27, 2021

Could you tell me the command you used to run training?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants