如何修改batchsize？

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

如何修改batchsize？ #9

Closed LZHhAG closed 1 week ago

LZHhAG commented 1 week ago

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