Seems like i am getting an error:
 File "mnist_film.py", line 299, in <module>              
 _, b_param = next(shiftLayer.named_parameters())                                                                               
StopIteration