very strange… I am still getting an error in the prediction step…
AttributeError: 'NoneType' object has no attribute 'detach'

I cants think what may be the issue and perhaps im thinking its related to class_weights which I used torch.from_numpy() on