Изменить ввод слоя во время сквозного обучения в pytorch - PullRequest
0 голосов
/ 16 апреля 2020

Я новичок в Pytorch, и я хотел бы разработать следующую модель:

enter image description here

Строительный блок "Generate Graph" не является частью сеть, и он просто генерирует график, используя функции f. Я также хочу, чтобы функция потерь была сквозной.

На самом деле я знаю, как спроектировать сеть из двух отдельных частей (Encoder и GCN), но я не могу связать их вместе с одной функцией потерь с graph_generator() между ними!

...