You are viewing a plain text version of this content. The canonical link for it is here.

Posted to github@beam.apache.org by GitBox <gi...@apache.org> on 2022/09/02 13:48:29 UTC

[GitHub] [beam] AnandInguva commented on issue #21450: Investigate load state_dict vs loading whole model

AnandInguva commented on issue #21450:
URL: https://github.com/apache/beam/issues/21450#issuecomment-1235528568

   No, thats a different solution on how to load the model_weights on the runner instead of SDK. that solution still needs the Model class python code.
   
   This issue references on how to load the model without the need of model class. But as of now, there is no recommended solution in Pytorch for this. Please look at https://pytorch.org/tutorials/beginner/saving_loading_models.html#save-load-entire-model


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: github-unsubscribe@beam.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org