DL-Art-School/codes/models/pixel_level_contrastive_learning
James Betker d1007ccfe7 Adjustments to pixpro to allow training against networks with arbitrarily large structural latents
- The pixpro latent now rescales the latent space instead of using a "coordinate vector", which
   **might** have performance implications.
- The latent against which the pixel loss is computed can now be a small, randomly sampled patch
   out of the entire latent, allowing further memory/computational discounts. Since the loss
   computation does not have a receptive field, this should not alter the loss.
- The instance projection size can now be separate from the pixel projection size.
- PixContrast removed entirely.
- ResUnet with full resolution added.
2021-01-12 09:17:45 -07:00
..
__init__.py Add lucidrains pixpro trainer 2021-01-05 20:14:22 -07:00
pixpro_lucidrains.py Adjustments to pixpro to allow training against networks with arbitrarily large structural latents 2021-01-12 09:17:45 -07:00
resnet_unet_2.py Adjustments to pixpro to allow training against networks with arbitrarily large structural latents 2021-01-12 09:17:45 -07:00
resnet_unet.py Allow uresnet to use pretrained resnet50 2021-01-10 12:57:31 -07:00