首先,主函数是if_cnn_mnist_work.py
1.输出snn测试结果的就是这么一些代码:
utils.pytorch_ann2snn(model_name=model_name,
norm_tensor=norm_tensor,
test_data_loader=test_data_loader,
device=device,
T=T,
log_dir=log_dir,
config=config
)
2.ctrl+鼠标左键点击pytorch_ann2snn
可以看到里面的代码
def pytorch_ann2snn(model_name, norm_tensor, test_data_loa