Traceback (most recent call last):


 "run_bertology.py", line 427, in <module>

main()


 "run_bertology.py", line 422, in main

head_mask = mask_heads(args, model, eval_dataloader)


 "run_bertology.py", line 180, in mask_heads

args, model, eval_dataloader, compute_entropy=False, head_mask=new_head_mask


 "run_bertology.py", line 105, in compute_heads_importance

head_importance += head_mask.grad.abs().detach()
AttributeError: 'NoneType' object has no attribute 'abs'

Traceback (most recent call last):


 "train.py", line 113, in <module>

tf.app.run()


 "/opt/anaconda2/envs/tf121py35/lib/python3.5/site-packages/tensorflow/python/platform/app.py", line 48, in run

_sys.exit(main(_sys.argv[:1] + flags_passthrough))


 "train.py", line 30, in main

FLAGS.batch_image_size, FLAGS.stride_size)


 "/data1/xxx/cv/dcscn-super-resolution/DCSCN.py", line 160, in load_datasets

datasets.build_batch(data_dir, batch_dir)


 "/data1/xxx/cv/dcscn-super-resolution/helper/loader.py", line 185, in build_batch

input_count = input_batch_images.shape[0]
AttributeError: 'NoneType' object has no attribute 'shape'
