Comments (1)
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)
- Poor performance đ after training on custom data âšī¸ HOT 1
- Can I run train.py ignoring characters that are in the dataset but not in the --character? HOT 1
- When ground truth is incorrect but, prediction is well-predicted. HOT 3
- Training stuck after first iteration
- rtmp://live.restream.io/live
- training success ,but failed is run as python file
- training success , but run python code error, about
- Text Color Recognition in Images
- after Fine tuning, demo.py can't decode original data HOT 1
- how to get coordinate of the text
- how i convert .zip files to .pth?
- AssertionError: datasets should not be an empty iterable
- CUDA error, while training Attn as prediction
- Cannot train model for upper case characters
- I encountered a bug when executing the following command in the deep-text-recognition-benchmark repository HOT 1
- [Question] Configuring the dataset for custom training HOT 2
- custom dataset Calculated padded input size per channel: (1 x 22). Kernel size: (2 x 2)
- Check the wrong prediction data
- Can it be used to extract texts from an entire page ?
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
đ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
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.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google â¤ī¸ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from deep-text-recognition-benchmark.