for i, data in enumerate(dataloader, 0):
netGDN.train()
iter_start_time = time.time()
if total_steps % 100 == 0:
t_data = iter_start_time - iter_data_time
visualizer.reset()
total_steps += opt.batchSize
epoch_iter += opt.batchSize
input, target, name = data
batch_size = target.size(0)
input = input.cuda()
target = target.cuda()
optimizerG.zero_grad()
oups = netGDN(input)
vgg_target = vgg19(target)
feat_target = vgg_target['conv3_2']
CX_loss_list = [criterionCX(vgg19(x_hat)['conv3_2'] ,feat_target) for x_hat in oups]
loss_CX = CX_loss_list[0]
L = loss_CX
L.backward()
optimizerG.step()
ganIterations += 1
are there some errors in the code?
for i, data in enumerate(dataloader, 0):
are there some errors in the code?