增加fast_split_to_train_test_set函数

fangwei123456 / spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

https://spikingjelly.readthedocs.io

Other

1.4k stars 246 forks source link

Closed TTiannaiTT closed 1 week ago

TTiannaiTT commented 1 week ago

运行n_caltech101和CIFAR10-DVS的时候训练集和测试集划分非常耗时间，故修改原来的split_to_train_test_set为多进程版本，加快了速度，函数定义尽量按照原来的设计，增加了batchsize设置单次处理的批量大小