Skip to content

Commit 0aeb222

Browse files
committed
Fix that the pretrained model doens't have use_att option.
1 parent b58e41d commit 0aeb222

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dataloader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def __init__(self, opt):
3939
self.opt = opt
4040
self.batch_size = self.opt.batch_size
4141
self.seq_per_img = opt.seq_per_img
42-
self.use_att = opt.use_att
42+
self.use_att = getattr(opt, 'use_att', True)
4343

4444
# load the json file which contains additional information about the dataset
4545
print('DataLoader loading json file: ', opt.input_json)

0 commit comments

Comments
 (0)