Traceback (most recent call last):


 "inference.py", line 124, in <module>

main()


 "inference.py", line 68, in main

checkpoint_path=args.checkpoint)


 "/ssd/pytorch/pytorch-image-models/timm/models/factory.py", line 42, in create_model

load_checkpoint(model, checkpoint_path)


 "/ssd/pytorch/pytorch-image-models/timm/models/helpers.py", line 22, in load_checkpoint

model.load_state_dict(new_state_dict)


 "/home/ivan/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 839, in load_state_dict

self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for GenEfficientNet:

Traceback (most recent call last):


 "E:\python361\Lib\site-packages\sru\sru_functional.py", line 17, in <module>

sru_cpu_impl = load(name="sru_cpu_impl", sources=[cpu_source])


 "E:\python361\lib\site-packages\torch\utils\cpp_extension.py", line 514, in load

with_cuda=with_cuda)


 "E:\python361\lib\site-packages\torch\utils\cpp_extension.py", line 682, in _jit_compile

_build_extension_module(name, build_directory)


 "E:\python361\lib\site-packages\torch\utils\cpp_extension.py", line 765, in _build_extension_module

name, error.output.decode()))
RuntimeError: Error building extension 'sru_cpu_impl': [1/2] cl /showIncludes -DTORCH_EXTENSION_NAME=sru_cpu_impl -IE:\python361\lib\site-packages\torch\lib\include -IE:\python361\lib\site-packages\torch\lib\include\TH -IE:\python361\lib\site-packages\torch\lib\include\THC -IE:\python361\Include -D_GLIBCXX_USE_CXX11_ABI=0 -fPIC -std=c++11 -c E:\python361\Lib\site-packages\sru\sru_cpu_impl.cpp /Fosru_cpu_impl.o
