chineseocr踩雷

在尝试chineseocr时，git地址后运行setup.md文件安装环境

修改torch版本为0.4.0 torchvision为0.2.2

然后根据提示pip各种包之后出现了
RuntimeError: Error(s) in loading state_dict for CRNN:Unexpected key(s) in state_dict: "cnn.batchnorm4

.num_batches_tracked", "cnn.batchnorm6.num_batches_tracked".

百度之后，有的说是torch版本问题，，0.4.0 但是运行了0.4.1，但是我重新安的就是0.4.0

最终：进入modules目录，找到batchnorm.py文件 vim打开

cd cd anaconda3/envs/chinesrocr/lib/python3.6/site-packages/torch/nn/modules

vim batchnorm.py

在第28行添加：

self.register_buffer('num_batches_tracked',torch.LongTensor([0]))#fix

在32行添加：

self.register_buffer('num_batches_tracked',None)#fix

修改后代码如下：

        if self.track_running_stats:
            self.register_buffer('running_mean', torch.zeros(num_features))
            self.register_buffer('running_var', torch.ones(nu