fangwei123456/spikingjelly

请问框架中的tdBN方法是还没有完成吗？

Closed this issue · 7 comments

QgZhan commented

@Yanqi-Chen
@fangwei123456

SpikingJelly version

0.0.0.0.14

Description

activation_based.layer.ThresholdDependentBatchNorm1d
activation_based.layer.ThresholdDependentBatchNorm2d
activation_based.layer.ThresholdDependentBatchNorm3d
三个类型似乎没有提供具体的实现？

fangwei123456 commented

直接使用就行，都继承自 _ThresholdDependentBatchNormBase。正如pytorch中的BN123D都继承自同一个基类

WeekendCheung commented

@fangwei123456

SpikingJelly version

0.0.0.0.14

Description
您好我也遇到了同样的问题，似乎是_check_input_dim()在子类没有实现导致了报NotImplementedError

gshaowei6 commented

@fangwei123456 _ThresholdDependentBatchNormBase类有一个bug，就是它继承了基类，但是没有实现基类里面得_check_input_dim函数，麻烦进行修复这个bug
def _check_input_dim(self, input): if input.dim() != 2 and input.dim() != 3 and input.dim() != 4 and input.dim() != 5: raise ValueError( f"expected 2D or 3D input (got {input.dim()}D input)" )

fangwei123456 commented

修复了_ThresholdDependentBatchNormBase的这个问题

gshaowei6 commented

提醒一下就是目前您修改之后这个bug依然存在，需要修复_ThresholdDependentBatchNormBase这个类，这个类里面没有实现_check_input_dim。

fangwei123456 commented

从GitHub安装最新版框架试试呢？

gshaowei6 commented

安装最新版本后没有问题了，谢谢