برای ذخیره مدل کافی است torch.save(model.state_dict(), 'model.pth') را اجرا کنی. برای بارگذاری نیز از torch.load و load_state_dict استفاده کن. همیشه نسخه PyTorch مدل ذخیرهشده را با محیط فعلی چک کن.
برای مشاهدهٔ ادامه، خرید کنید
دسترسی سریع و فوری