traiNNer-redux documentation
traiNNer-redux is a deep learning training framework for image super resolution and restoration which allows you to train PyTorch models for upscaling and restoring images and videos. NVIDIA graphics card is recommended, but AMD works on Linux machines with ROCm.
Reference
- Config file reference
- Architecture reference
- Loss reference
- adistsloss
- aesoploss
- averageloss
- bcewithlogitsdiceloss
- bicubicloss
- charbonnierloss
- colorloss
- contextualloss
- cosimloss
- distsloss
- ffloss
- fftloss
- fliploss
- ganloss
- gradientvarianceloss
- hsluvloss
- l1loss
- ldlloss
- linedistillerloss
- lumaloss
- mseloss
- mssimloss
- msssiml1loss
- multiscaleganloss
- nccloss
- psnrloss
- perceptualanimeloss
- perceptualfp16loss
- perceptualloss
- ssimloss
Benchmarks