Softmax (dim=1) ) def forward(self, x): return self.net(x) The model we're ... net): obs = env.reset() reward = 0.0 steps = 0 while True: obs_v = torch.
確定! 回上一頁