RuntimeError: Error(s) in loading state_dict for CRAFT:

[satyam-pivotchain]

i have train detection model (CRAFT) for my custom dataset and in exp/custom_dataset folder i got weights ( train on google colab using GPU). when i am trying to use those train weights in my system with EasyOCR i got errors like this

RuntimeError: Error(s) in loading state_dict for CRAFT: Missing key(s) in state_dict: "basenet.slice1.0.weight", "basenet.slice1.0.bias", "basenet.slice1.1.weight" Etc Unexpected key(s) in state_dict: "iter", "craft", "optimizer", "scaler".

i have used the following code

import easyocr
import os
import pandas as pd

fol_path = "/home/satyam/PycharmProjects/PivotProject/NP_Images"
## reader = easyocr.Reader(['en'], model_storage_directory="/home/satyam/PycharmProjects/Custom_EasyOcr/model")
reader = easyocr.Reader(['en'])

[ghost]

I have make Changes in config.py

[rkcosmos]

@gmuffiness

[Ali-Fayzi]

Hi

you can load like this :

torch.load('./CRAFT.pth')['craft']

[Shidhani]

I have make Changes in config.py

when i try this I get RuntimeError: Error(s) in loading state_dict for CRAFT: Missing key(s) in state_dict: "basenet.slice1.0.weight", "basenet.slice1.0.bias", "basenet.slice1.1.weight", ..