1.訓練過程中學習率衰減
if (self.e+1) > (self.num_epochs - self.num_epochs_decay):g_lr -= (self.g_lr / float(self.num_epochs_decay))d_lr -= (self.d_lr / float(self.num_epochs_decay))self.update_lr(g_lr, d_lr)print('Decay learning rate to g_lr: {}, d_lr:{}.'.format(g_lr, d_lr))
def update_lr(self, g_lr, d_lr):for param_group in self.g_optimizer.param_groups:param_group['lr'] = g_lrfor param_group in self.d_A_optimizer.param_groups:param_group['lr'] = d_lrfor param_group in self.d_B_optimizer.param_groups:param_group['lr'] = d_lr
，torch.cuda.device_count()，torch.cuda.get_device_name(0))
)
)
)
時間單位(ms μs ns ps)長度單位(dm cm mm μm nm pm fm am zm ym))
