cyto_dl.nn.head package

Submodules

• cyto_dl.nn.head.base_head module
  • BaseHead
    • BaseHead.forward()
    • BaseHead.generate_io_map()
    • BaseHead.run_head()
    • BaseHead.update_params()
• cyto_dl.nn.head.gan_head module
  • GANHead
    • GANHead.forward()
    • GANHead.run_head()
• cyto_dl.nn.head.gan_head_superres module
  • GANHead_resize
    • GANHead_resize.forward()
• cyto_dl.nn.head.mae_head module
  • MAEHead
    • MAEHead.run_head()
• cyto_dl.nn.head.mask_head module
  • MaskHead
    • MaskHead.run_head()
• cyto_dl.nn.head.res_blocks_head module
  • ResBlocksHead
    • ResBlocksHead.forward()
• cyto_dl.nn.head.vic_reg module
  • VICRegHead
    • VICRegHead.run_head()