-
相同的parser在train.py文件夹下可以正常运行,但在ann_to_snn.py下报错。
我的数据集图像大小为800*800,修改下列parser为:
parser.add_argument('--img-size', nargs='+', type=int, default=[800, 800], help='[min_train, max-train, test]')
会报错V…
-
when I try to transform ann to snn ,some problems occurred:
$ python3 ann_to_snn.py --cfg cfg/yolov3-tiny.cfg --data data/coco.data --weights weights/best.pt --timesteps 128Namespace(augment=False,…
-
D:\ANACONDA\envs\Mytorch\python.exe D:/SNN/PyTorch-Spiking-YOLOv3-ultralytics/ann_to_snn.py
Namespace(augment=False, batch_size=2, cache_images=False, cfg='cfg/yolov3-spp.cfg', channel_wise=False, co…
-
Dear authors,
Congratulations on your acceptance to ICML 2024! I am the first author of SpikeBERT, and I am very grateful for your citation of our work. I am also delighted to see the integration o…
-
你好,
非常感谢你的源码,解决了我对Spiking YOLO的很多疑惑。
但我在运行ann_to_snn.py的过程中遇到了内存开销过高的问题,导致device='cpu'时内存占用将近80G,而无法在GPU上运行。
我发现在调用例如:
https://github.com/cwq159/PyTorch-Spiking-YOLOv3/blob/941c3faddc2ecc6ab02086a…
-
Hello,
I am trying to run your QBOX and other repos as this is very interesting but there seems to be missing the custom CUBE_REPORT import:
```
import CUBE_REPORT # Import the custom report m…
-
### 🚀 The feature, motivation and pitch
moving from [slack discussion](https://pytorch.slack.com/archives/C3PDTEV8E/p1709153741504309)
How can FSDP shard a model directly to GPUs during model i…
-
Hi,
**Describe the issue**
I am using ONNX Runtime python api for inferencing, during which the memory is spiking continuosly.
(Model information - Converted pytorch based transformers model t…
-
### Model description
**Abstract:**
>As the size of large language models continue to scale, so does the computational resources required to run it. Spiking neural networks (SNNs) have emerged as an…
-
I'm trying to fine-tune BGE-M3 based on the README here: https://github.com/FlagOpen/FlagEmbedding/tree/master/examples/finetune
I originally started with the latest transformers version a few week…