Я нашел только источники о том, как это сделать. получите модель torchscript, работающую на Android, но не чистую модель Pytorch. Поэтому я попытался преобразовать модель в torchscript, но безуспешно.
Я пробовал:
- но модель, похоже, сильно зависит от входных данных, и поэтому ее отслеживание прерывается в тот момент, когда я использую разные входные данные.
Код: Выделить всё
torch.jit.trace(model, example) - но не смог заставить его работать, потому что кажется, что каждая вторая строка использует функцию, несовместимую с функцией скрипта.
Код: Выделить всё
torch.jit.script(model) - работал при компиляции, но я понятия не имею, как сохранить скомпилированную модель или загрузить ее. Так что этот тоже не работает.
Код: Выделить всё
torch.compile(model)
Есть идеи, как заставить это работать?
Подробнее здесь: https://stackoverflow.com/questions/785 ... ng-flutter