!540 fix bugs for imagenet

Merge pull request !540 from moran/master

mindinsight/wizard/conf/templates/network/alexnet/scripts/run_eval.sh-tpl

@@ -61,5 +61,5 @@ cp -r ../src ./eval
 cd ./eval || exit
 env > env.log
 echo "start evaluation for device $DEVICE_ID"
-python eval.py --dataset_path=$PATH1 --checkpoint_path=$PATH2 &> log &
+python eval.py --dataset_path=$PATH1 --checkpoint_path=$PATH2 --dataset_sink_mode=False &> log &
 cd ..

mindinsight/wizard/conf/templates/network/resnet50/scripts/run_eval.sh-tpl

@@ -61,5 +61,5 @@ cp -r ../src ./eval
 cd ./eval || exit
 env > env.log
 echo "start evaluation for device $DEVICE_ID"
-python eval.py --dataset_path=$PATH1 --checkpoint_path=$PATH2 &> log &
+python eval.py --dataset_path=$PATH1 --checkpoint_path=$PATH2 --dataset_sink_mode=False &> log &
 cd ..

mindinsight/wizard/conf/templates/network/resnet50/train.py-tpl

@@ -124,7 +124,7 @@ if __name__ == '__main__':
         {% if dataset=='ImageNet' %}
         if not cfg.use_label_smooth:
             cfg.label_smooth_factor = 0.0
-        loss = CrossEntLambropy(smooth_factor=cfg.label_smooth_factor, num_classes=cfg.num_classes)
+        loss = CrossEntropy(smooth_factor=cfg.label_smooth_factor, num_classes=cfg.num_classes)
         {% else %}
         {% if loss=='SoftmaxCrossEntropyWithLogits' %}
         loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction='mean')