[6] Ways to save the neural network in PyTorch

Поділитися
Вставка
  • Опубліковано 27 гру 2024

КОМЕНТАРІ • 10

  • @bohdanmelnychenko2865
    @bohdanmelnychenko2865 4 роки тому +1

    Cheers, saved my assignment, lad xx

  • @iramarshad700
    @iramarshad700 3 роки тому

    Hi, can you share the Jupyter notebook link?

    • @ifelsestatement7803
      @ifelsestatement7803  3 роки тому

      Hi! Unfortunately, I have deleted it. Sorry

    • @iramarshad700
      @iramarshad700 3 роки тому

      @@ifelsestatement7803 I tried it but I am only able to save last training epoch and accuracy. I want to save all epochs along with accuracy and loss. Can you help me?

    • @ifelsestatement7803
      @ifelsestatement7803  3 роки тому +1

      @@iramarshad700 Sure. But why would you want to save all of them? Just save the best one. But if you still want to save all of them. Just save a model on each iteration of each epoch and give a unique name to the model. Something like: model + epoch_number. To get like "model1", "model2", "model3", etc. So that each file would represent each epoch. Just when you save the model as a file and you give the already existing file name, it overrides the previous file.

    • @iramarshad700
      @iramarshad700 3 роки тому +1

      @@ifelsestatement7803 I am learning how to save a model and how I will use it further. So I though may be I can save all the checkpoints and I dont need to run train data again and again

    • @ifelsestatement7803
      @ifelsestatement7803  3 роки тому

      @@iramarshad700 ok, I see now. Did my suggestion help you? All good now?

  • @SHREERAJM
    @SHREERAJM Рік тому

    please if anyone have this project jupyter note book please shatre me'