python convert_tf2_h5.py \
  /data/xch/chm_workspace2/Models/starnet_weights2/weights_D_Greyscale.h5 \
  /data/xch/chm_workspace2/Models/starnet_weights2/weights_D_Greyscale.pth \
  --mode Greyscale



  # /data/xch/chm_workspace2/Models/starnet_weights2/weights_G_RGB.h5 \
  # /data/xch/chm_workspace2/Models/starnet_weights2/weights_G_RGB.pth \

#   /data/xch/chm_workspace2/Models/starnet_weights2/weights_G_Greyscale.h5 \
#   /data/xch/chm_workspace2/Models/starnet_weights2/weights_G_Greyscale.pth \
#   --mode Greyscale