DQN tutorial run error

reinforcement-learning
1

Getting this error in running DQN tutorial.

Traceback (most recent call last):
File “reinforcement_q_learning.py”, line 109, in
plt.imshow(get_screen().cpu().squeeze(0).permute(1, 2, 0).numpy(),
File “reinforcement_q_learning.py”, line 105, in get_screen
return resize(screen).unsqueeze(0).type(Tensor)
NameError: global name ‘Tensor’ is not defined

2

Tensor is defined in the first code snippet:

# if gpu is to be used
use_cuda = torch.cuda.is_available()
FloatTensor = torch.cuda.FloatTensor if use_cuda else torch.FloatTensor
LongTensor = torch.cuda.LongTensor if use_cuda else torch.LongTensor
ByteTensor = torch.cuda.ByteTensor if use_cuda else torch.ByteTensor
Tensor = FloatTensor
3

It worked thank you. I had commented some part of the code and this commented this also.