Unable to return a boolean variable from Pytorch Dataset’s __get_item__
I have a pytorch Dataset subclass and I create a pytorch DataLoader out of it. It works when I return two tensors from DataSet’s __getitem__() method. I tried to create minimal (but not working, more on this later) code as below:
Unable to return a boolean variable from Pytorch Dataset’s __get_item__
I have a pytorch Dataset subclass and I create a pytorch DataLoader out of it. It works when I return two tensors from DataSet’s __getitem__() method. I tried to create minimal (but not working, more on this later) code as below: