GuanHengcong

> 1. SFE模块的unfold可以用如下模块代替，可以减少很多计算子 > > ``` > import torch > import torch.nn as nn > class Unfold(nn.Module): > def __init__(self): > super().__init__() > kernel = torch.eye(3) > kernel = kernel.view(3,...

> @GuanHengcong 因为重构以后的Unfold它的名称就和原来state_dict里面的对不上了，最简单的办法就是把convert_to_stream里面的 > > ``` > else: > raise (....) > ``` > > 给注释了 以及SFE的后面也不用reshape 大佬，很抱歉没有及时回复您消息，我把reshape和else都去掉了，报错如下，您可以再帮忙看看吗    在“torch.onnx.export(”导出这行报错

> @GuanHengcong 整体模型的里面的那个SFE的channel数是3，你把那个换成 > > ``` > import torch > import torch.nn as nn > class Unfold_in(nn.Module): > def __init__(self): > super().__init__() > kernel = torch.eye(3) > kernel = kernel.view(3,...