ChenRocks/UNITER

AttributeError: 'Namespace' object has no attribute 'train_datasets'

ForawardStar opened this issue · 2 comments

When I launched the 2nd stage training using :

horovodrun -np 4 python pretrain_vcr.py --config config/pretrain-vcr-base-4gpu.json \
    --output_dir $PRETRAIN_VCR_EXP

An error occured, which is:

Traceback (most recent call last):

File "pretrain_vcr.py", line 560, in <module>

main(args)

File "pretrain_vcr.py", line 205, in main

all_dbs = [db for datasets in [opts.train_datasets, opts.val_datasets]

AttributeError: 'Namespace' object has no attribute 'train_datasets'

How to fix this problem? Thanks.

train_datasets is indeed in opts (which is loaded from config)

"train_datasets": [

Thanks for your reply! It is helpful to me. Here I want to launch the first stage pre-training on COCO, Visual Genome, CC, and SBU. Can your provide the linking for downloading CC, and SBU? If their sizes are too large, can you simply provide the .db file of text input, or the scripts for producing such .db file?