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