Sayed_Nadim:

.type(torch.IntTensor)


on an another note, is it not a good an idea to train a one class detection model with RGB values or I should follow the good practice of scaling and normalizing my tensors?