You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The structure of data directory is:
dataset
├── annotations
│ ├── captions_train2014.json
│ ├── captions_val2014.json
│ ├── coco_dev_ids.npy
│ ├── IIIT5k_3000
│ ├── coco_restval_ids.npy
│ ├── coco_test_ids.npy
│ └── coco_train_ids.npy
├── feature
│ ├── coco_all_align.hdf5
And to train your model with the parameters used in our experiments, use:
CUDA_VISIBLE_DEVICES=0,1 python train.py --exp_name m2_transformer --batch_size 8 --m 40 --head 8 --warmup 10000 --features_path dataset/feature/coco_all_align.hdf5 --annotation_folder dataset/annotations
I get the error as follow:
Why did this error happen? How to solve it?Thank you!
The text was updated successfully, but these errors were encountered:
The structure of data directory is:
dataset
├── annotations
│ ├── captions_train2014.json
│ ├── captions_val2014.json
│ ├── coco_dev_ids.npy
│ ├── IIIT5k_3000
│ ├── coco_restval_ids.npy
│ ├── coco_test_ids.npy
│ └── coco_train_ids.npy
├── feature
│ ├── coco_all_align.hdf5
And to train your model with the parameters used in our experiments, use:
CUDA_VISIBLE_DEVICES=0,1 python train.py --exp_name m2_transformer --batch_size 8 --m 40 --head 8 --warmup 10000 --features_path dataset/feature/coco_all_align.hdf5 --annotation_folder dataset/annotations
I get the error as follow:
Why did this error happen? How to solve it?Thank you!
The text was updated successfully, but these errors were encountered: