Same Error, my torch version is 1.3.1
Traceback (most recent call last):
  File "source/scripts/build_activation_images.py", line 68, in <module>
    function = torch.nn.modules.activation.__dict__[function_name]()
KeyError: 'GELU'
make: *** [figures] Error 1