Hello,
Pytorch doesn’t accept string classes. So, l converted my labels to int and feed my network as follow :
However l need to keep track of my real labels (string names). I need at the end of my learning get the perfomance on each example and map the int label to string label.
How can l do that ?
with open('/home/train.sav', 'rb') as handle:
train_data = pickle.load(handle)
train_loader = torch.utils.data.DataLoader(
train_data, batch_size=args.batch_size, shuffle=(train_sampler is None),
num_workers=args.workers, pin_memory=True, sampler=train_sampler)
for i, (input, target) in enumerate(train_loader):
target = target.cuda(async=True)
input_var = torch.autograd.Variable(input)
target_var = torch.autograd.Variable(target)
Thank you