Behrad Koohy

icon indicating a website link https://behradkoohy.github.io/

icon company @southampton icon location Southampton, UK icon location PhD Computer Science candidate at University of Southampton

Results 1 issues of Behrad Koohy

RuntimeError: Size does not match at dimension 0 expected index [32, 1] to be smaller than self [1, 3] apart from dimension 1

1 comment

Hi, I'm trying to setup a DQN agent with a graph attention layer. The agent can take one of 3 actions. For some reason, when I run the training function,...