search

Media-Smart / vedastr

A scene text recognition toolbox based on PyTorch
Apache License 2.0
535 stars 100 forks source link

Error: __init__() got an unexpected keyword argument 'seed' #67

Closed PhamLeQuangNhat closed 3 years ago

PhamLeQuangNhat commented 3 years ago

I run tools/dist_train.sh configs/cstr.py 4 and error:

File "tools/train.py", line 49, in main() File "tools/train.py", line 44, in main runner = TrainRunner(train_cfg, inference_cfg, common_cfg) File "tools/../vedastr/runners/train_runner.py", line 18, in init train_cfg['data']['train']) File "tools/../vedastr/runners/base.py", line 105, in _build_dataloader dict(dataset=dataset)) File "tools/../vedastr/dataloaders/samplers/builder.py", line 7, in build_sampler sampler = build_from_cfg(cfg, DISTSAMPLER, default_args) File "tools/../vedastr/utils/common.py", line 13, in build_from_cfg return obj_from_dict_registry(cfg, parent, default_args) File "tools/../vedastr/utils/common.py", line 78, in obj_from_dict_registry return obj_cls(args) File "tools/../vedastr/dataloaders/samplers/dist_balance_sampler.py", line 77, in init super(BalanceSampler, self).init(dataset=dataset, kwargs) TypeError: init() got an unexpected keyword argument 'seed'

How to solve this problem? Thank in advance.

ChaseMonsterAway commented 3 years ago

Hi, please use PyTorch whose version is larger than 1.6.0. ( I will update the requirements. )