problem with classification

[alibeknakhimov]

import torch
from torch.utils.mobile_optimizer import optimize_for_mobile

model = torch.load('best.pt',map_location="cpu")
model.eval()
example = torch.rand(1, 3, 224, 224)
traced_script_module = torch.jit.trace(model, example)
optimized_traced_model = optimize_for_mobile(traced_script_module)
optimized_traced_model._save_for_lite_interpreter("model.pt")

when i use this code for optimize for mobile it has error

AttributeError Traceback (most recent call last)
in <cell line: 6>()
4
5 model = torch.load('best.pt',map_location="cpu")
----> 6 model.eval()
7 example = torch.rand(1, 3, 224, 224)
8 traced_script_module = torch.jit.trace(model, example)

AttributeError: 'dict' object has no attribute 'eval'