Преобразование онлайн-потока данных в torch_geometric.data.data.Data
У меня есть онлайн-поток поступающих данных, мне нужно их обработать и из каждой точки данных в моем пакете создать точку данных в стиле torch_geometric. Итак, если в моем пакете 128 точек, мне нужно преобразовать его в пакет из 128 графиков и передать его в мой GNN.
Мне нужно выполнить это, не создавая много помех в цикле обучения, поэтому перебирать каждый образец — последнее, что хочется делать. Я пробовал functorch vmap, но поскольку он возвращает нетензорный тип данных, я получаю сообщение об ошибке.
Любая помощь или идеи очень ценятся.