fix deprecated loss access (pytorch#113)

Addressing pytorch#108
sashaDoubov · Feb 3, 2019 · fb7709a · fb7709a
1 parent 511db2b
commit fb7709a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchnet/logger/meterlogger.py b/torchnet/logger/meterlogger.py
@@ -98,7 +98,7 @@ def update_loss(self, loss, meter='loss'):
         loss = self.__to_tensor(loss)
         if meter not in self.meter.keys():
             self.__addloss(meter)
-        self.meter[meter].add(loss[0])
+        self.meter[meter].add(loss.item())
 
     def peek_meter(self):
         '''Returns a dict of all meters and their values.'''