Explorar o código

fixed classifier report function to reduce GPU-Memory accesses

Dimitri Korsch %!s(int64=3) %!d(string=hai) anos
pai
achega
eb1184fd8f
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      cvmodelz/classifiers/base.py

+ 1 - 1
cvmodelz/classifiers/base.py

@@ -31,7 +31,7 @@ class Classifier(chainer.Chain):
 		self.model = model
 
 	def report(self, **values) -> None:
-		chainer.report(values, self)
+		chainer.report({key: float(chainer.as_array(val)) for key, val in values.items()}, self)
 
 	def enable_only_head(self) -> None:
 		self.model.disable_update()