(param)
| 139 | |
| 140 | |
| 141 | def convert_param_name(param): |
| 142 | resnet_param = {} |
| 143 | for k, v in six.iteritems(param): |
| 144 | try: |
| 145 | newname = name_conversion(k) |
| 146 | except Exception: |
| 147 | logger.error("Exception when processing caffe layer {}".format(k)) |
| 148 | raise |
| 149 | logger.info("Name Transform: " + k + ' --> ' + newname) |
| 150 | resnet_param[newname] = v |
| 151 | return resnet_param |
| 152 | |
| 153 | |
| 154 | if __name__ == '__main__': |
no test coverage detected
searching dependent graphs…