Skip to content

Commit

Permalink
fixed channel naming for new metric objects like ROCAUC (#35)
Browse files Browse the repository at this point in the history
  • Loading branch information
jakubczakon authored Jun 6, 2019
1 parent 3a7355a commit a361b29
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion neptunecontrib/monitoring/fastai.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def on_epoch_end(self, **kwargs):
metric_names = ['valid_last_loss'] + kwargs['metrics']
for metric_value, metric_name in zip(metric_values, metric_names):
metric_name = getattr(metric_name, '__name__', metric_name)
self._exp.send_metric(self._prefix + metric_name, float(metric_value))
self._exp.send_metric(self._prefix + str(metric_name), float(metric_value))

def on_batch_end(self, **kwargs):
self._exp.send_metric('{}last_loss'.format(self._prefix), float(kwargs['last_loss']))

0 comments on commit a361b29

Please sign in to comment.