我将激活函数relu改为swish，模型训练没没有问题，使用load_model（）加载模型却报错报错为：builtins.UnboundLocalError: local variable 'class_name' referenced before assignment.
https://blog.csdn.net/uwr44uouqcnsuqb60zk2/article/details/78334072