We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b58e41d commit 0aeb222Copy full SHA for 0aeb222
dataloader.py
@@ -39,7 +39,7 @@ def __init__(self, opt):
39
self.opt = opt
40
self.batch_size = self.opt.batch_size
41
self.seq_per_img = opt.seq_per_img
42
- self.use_att = opt.use_att
+ self.use_att = getattr(opt, 'use_att', True)
43
44
# load the json file which contains additional information about the dataset
45
print('DataLoader loading json file: ', opt.input_json)
0 commit comments