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)
https://arxiv.org/abs/2407.20708
GNU Affero General Public License v3.0
111 stars 6 forks source link

如何修改batchsize? #9

Closed LZHhAG closed 1 month ago

LZHhAG commented 1 month ago

image 多卡训练需要卡数的倍数,需要修改batchsize,请问这个batchsize在哪里修改呢?谢谢