|
|
x['lr'] = np.interp(ni, xi, [0.0, x['initial_lr'] * self.lf(self.epoch)])
|
|
x['lr'] = np.interp(ni, xi, [0.0, x['initial_lr'] * self.lf(self.epoch)])
|
|
|
images = images.to(self.device, non_blocking=True).float()
|
|
images = images.to(self.device, non_blocking=True).float()
|