def __init__(self, image_nr, image_resize, model_name): super().__init__() self.model_name = model_name self.ctx = neptune.Context() self.epoch_loss_averager = Averager() self.image_nr = image_nr self.image_resize = image_resize
def __init__(self, image_nr, image_resize, image_every, model_name, use_depth): super().__init__() self.model_name = model_name self.epoch_loss_averager = Averager() self.image_resize = image_resize self.image_every = image_every self.image_nr = image_nr self.use_depth = use_depth
def on_batch_end(self, metrics, *args, **kwargs): for name, loss in metrics.items(): loss = loss.data.cpu().numpy()[0] if name in self.epoch_loss_averagers.keys(): self.epoch_loss_averagers[name].send(loss) else: self.epoch_loss_averagers[name] = Averager() self.epoch_loss_averagers[name].send(loss) if self.batch_every and ((self.batch_id % self.batch_every) == 0): logger.info('epoch {0} batch {1} {2}: {3:.5f}'.format(self.epoch_id, self.batch_id, name, loss)) self.batch_id += 1
def on_batch_end(self, metrics, *args, **kwargs): for name, loss in metrics.items(): loss = loss.data.cpu().numpy()[0] if name in self.epoch_loss_averagers.keys(): self.epoch_loss_averagers[name].send(loss) else: self.epoch_loss_averagers[name] = Averager() self.epoch_loss_averagers[name].send(loss) self.ctx.channel_send(name, loss) self.batch_id += 1
def on_batch_end(self, metrics, *args, **kwargs): for name, loss in metrics.items(): loss = loss.data.cpu().numpy()[0] if name in self.epoch_loss_averagers.keys(): self.epoch_loss_averagers[name].send(loss) else: self.epoch_loss_averagers[name] = Averager() self.epoch_loss_averagers[name].send(loss) neptune.send_metric('{} batch {} loss'.format( self.model_name, name), x=self.batch_id, y=loss) self.batch_id += 1
def on_batch_end(self, metrics, *args, **kwargs): for name, loss in metrics.items(): loss = loss.cpu().item() if name in self.epoch_loss_averagers.keys(): self.epoch_loss_averagers[name].send(loss) else: self.epoch_loss_averagers[name] = Averager() self.epoch_loss_averagers[name].send(loss) self.ctx.channel_send('{} batch {} loss'.format( self.model_name, name), x=self.batch_id, y=loss) self.batch_id += 1
def __init__(self, model_name): super().__init__() self.model_name = model_name self.ctx = neptune.Context() self.epoch_loss_averager = Averager()