Git Product home page Git Product logo

Comments (1)

Teddy000Jung avatar Teddy000Jung commented on May 24, 2024

I added the "--workers" to execute the code to solve this issue, but a new error message appeared.


dataset_root: lmdb/training
opt.select_data: ['MJ', 'ST']
opt.batch_ratio: ['0.5', '0.5']

dataset_root: lmdb/training dataset: MJ
sub-directory: /MJ num samples: 1000
num total samples of MJ: 1000 x 1.0 (total_data_usage_ratio) = 1000
num samples of MJ per batch: 192 x 0.5 (batch_ratio) = 96

dataset_root: lmdb/training dataset: ST
sub-directory: /ST num samples: 1000
num total samples of ST: 1000 x 1.0 (total_data_usage_ratio) = 1000
num samples of ST per batch: 192 x 0.5 (batch_ratio) = 96

Total_batch_size: 96+96 = 192

dataset_root: lmdb/validation dataset: /
sub-directory: /MJ num samples: 1000
sub-directory: /ST num samples: 1000

No Transformation module specified
model input parameters 32 100 20 1 512 256 63 25 None VGG BiLSTM CTC
loading pretrained model from None-VGG-BiLSTM-CTC.pth
Traceback (most recent call last):
File "train.py", line 317, in
train(opt)
File "train.py", line 84, in train
model.load_state_dict(torch.load(opt.saved_model, map_location=torch.device('cpu')), strict=False)
File "C:\Users\user\anaconda3\envs\EasyOCR\lib\site-packages\torch\nn\modules\module.py", line 1672, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DataParallel:
size mismatch for module.Prediction.weight: copying a param with shape torch.Size([37, 256]) from checkpoint, the shape in current model is torch.Size([63, 256]).
size mismatch for module.Prediction.bias: copying a param with shape torch.Size([37]) from checkpoint, the shape in current model is torch.Size([63]).

How can I resolve this issue?

from deep-text-recognition-benchmark.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤ī¸ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.