Beacuse of the varying video size i am unable to form batches. So i am using normal SGD. I have a doubt regarding the state_dict. What is the difference between saving the trained model using torch.save() and saving the state_dict?
Thanks