#object #detection #meta
Вопрос:
у меня есть проблема для первого этапа обучения с этой командой и настройкой:
инструменты python3 -m.train_net —num-графические процессоры 1 —конфигурационный файл конфигурации/PascalVOC-обнаружение/split1/faster_rcnn_R_101_FPN_base1.yaml
Это последние три строки ошибки:
result = self.forward(*input, **kwargs) File "/home/cudaa/anaconda3/envs/fsod/lib/python3.6/site-packages/detectron2/layers/batch_norm.py", line 54, in forward eturn x * scale.to(out_dtype) bias.to(out_dtype) RuntimeError: CUDA out of memory. Tried to allocate 464.00 MiB (GPU 0; 5.94 GiB total capacity; 4.47 GiB already allocated; 94.38 MiB free; 4.93 GiB reserved in total by PyTorch)
я думаю, что проблема должна иметь отношение к размеру пакета, но как я могу изменить ее для первого этапа обучения? я не видел никакой конфигурации для этого…этот метод использует detectron2 для обнаружения.
Я благодарю всех, кто может помочь решить мою проблему
Комментарии:
1. Привет, Джон, не могли бы вы объяснить, что такое tools.train_net ?