Skip to content

MrVoid918/AnimeGAN-torch

Repository files navigation

AnimeGAN-torch

This repo is to replicate results from AnimeGANv2 in Pytorch

This branch is to implement NOGAN training from Deoldify. Since GAN training itself is unstable, it is desirable to have as minimal GAN training as possible yet still have the effects of GAN. NOGAN works well in image colourization, which is a task in image-to-image translation, intuitively it should also apply to other image-to-image translation tasks. Feature Loss of NOGAN training of AnimeGAN uses same loss in AnimeGAN. The training of NOGAN happens in notebooks where the training is moderated heavily, with a lot of checkpointing and trial and error.

This branch adds FP16 training form Nvidia Apex. Refer to the docs on different levels set to training. FP16 training is not applied to GAN training.

About

Reproducing Results from AnimeGANv2 with Pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published