def custom_setup(self):
# create loss meters
self.loss_meter_G = AverageMeter(name='Loss_G')
self.loss_meter_D = AverageMeter(name='Loss_D')
self.optical = ParamSet(name='optical', size=self.config.DATASET.optical_size, output_format=self.config.DATASET.optical_format)