d1007ccfe7
- 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. |
||
---|---|---|
.. | ||
__init__.py | ||
pixpro_lucidrains.py | ||
resnet_unet_2.py | ||
resnet_unet.py |