Fix CIFAR10 example

Author: 1adrianb

bnn/engine.py

@@ -3,7 +3,7 @@
 from easydict import EasyDict as edict
 from typing import List, Callable, Dict
 
-from . import BConfig, prepare_binary_model
+from . import BConfig, prepare_binary_model, Identity
 from .ops import *
 
 

examples/cifar10.py

@@ -14,7 +14,7 @@
     BasicInputBinarizer,
     XNORWeightBinarizer
 )
-from bnn import BConfig, prepare_binary_model
+from bnn import BConfig, prepare_binary_model, Identity
 from  bnn.models.resnet import resnet18
 
 from .utils import AverageMeter, ProgressMeter, accuracy
@@ -64,7 +64,7 @@
 print('==> Preparing the model for binarization')
 bconfig = BConfig(
             activation_pre_process = BasicInputBinarizer,
-            activation_post_process = nn.Identity,
+            activation_post_process = Identity,
             weight_pre_process = XNORWeightBinarizer
         )
 # first and last layer will be kept FP32

examples/recepies/xnor-net.yaml

@@ -3,7 +3,7 @@ step0:
   pre_activation:
     name: "BasicInputBinarizer"
   post_activation:
-    NAME: "nn.Identity"
+    NAME: "Identity"
   weight:
     name: "XNORWeightBinarizer"
     args: