Hi @ptrblck,
Thanks for your code but I got the following error,
Traceback (most recent call last):
  File "train.py", line 32, in <module>
    model.optimize_parameters()
  File "/home/csdept/projects/pytorch-CycleGAN-and-pix2pix/models/pix2pix_model.py", line 209, in optimize_parameters
    self.forward()
  File "/home/csdept/projects/pytorch-CycleGAN-and-pix2pix/models/pix2pix_model.py", line 112, in forward
    self.model_feature.unet_block_1.register_forward_hook(self.get_activation('block1'))
TypeError: get_activation() takes 1 positional argument but 2 were given