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()
