tensor- все статьи тега ➜ страница 0


Лучший способ сохранить обученную модель в PyTorch?

Я искал альтернативные способы сохранить обученную модель в PyTorch. До сих пор я нашел две альтернативы. факел.сохранить() для сохранения модели и факел.нагрузки() для загрузки модели. модель.state_dict () для сохранения обученной модели и модель.load_state_dict () для загрузки сохраненной модели. я наткнулся на это обсуждение где подход 2 рекомендуется за 1 подход. мой вопрос в том, почему предпочтителен второй подход? Это только потому, что факел.nn модули имеют эти две функции, и нам р ...