code: def on_epoch_end(self, epoch, *args, **kwargs):
      x = self.model.predict(np.random.normal(size=[8, 128]))
      x = np.concatenate(list(x * 255), axis=1).astype(np.uint8)
      img = PIL.Image.fromarray(x)
      img.save(os.path.join(self.path, f"epoch-{epoch:04d}.png"))
  
file_path: elegy/examples/need-fixing/WGAN-GP/main.py
imports: import glob
  import os
  import pickle
  import numpy
  import PIL.Image
  from absl import app, flags
  from model import WGAN_GP
  import elegy
label_desc: Use image node_id 2 for calling the SaveImagesCallback obj's underlying member method code with example usage: obj.on_epoch_end(epoch) without return types