forked from ecker/DL-Art-School
- 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 | ||