search

BICLab / SpikeYOLO

Offical implementation of "Integer-Valued Training and Spike-Driven Inference Spiking Neural Network for High-performance and Energy-efficient Object Detection" (ECCV2024 Best Paper Candidate / Oral)
https://arxiv.org/abs/2407.20708
GNU Affero General Public License v3.0
97 stars 4 forks source link

COCO2017训练结果与论文中不同？ #11

Open LZHhAG opened 3 days ago

LZHhAG commented 3 days ago

作者您好，我按照您给的代码配置了环境，用COCO2017数据集进行训练，但训练结果和您在论文中给的结果有差异，不知道您这套代码，默认设置的T和D都是多少呢？ image image

XinhaoLuo666 commented 15 minutes ago

您好，代码默认的T和D应该是1和4。我怀疑您的训练没有完全收敛，请问您训练了多少epoch呢？我们论文中训练了300ep。

LZHhAG commented 12 minutes ago

我训练了100个epoch，看来是要再训练一下了，谢谢您的回答。另外您代码中默认设置的T和D在哪里可以修改呢？我是初学者，确实还没有找到。望您赐教，谢谢。