add more loss functions and optimizers

LumenPallidium / neural-file-sorter

A neural network based file sorter. Trains an autoencoder to sort images or audio based on the similarity of their encodings, or uses the OpenAI CLIP model.

MIT License

27 stars 1 forks source link

add more loss functions and optimizers #5

Open LumenPallidium opened 3 years ago

LumenPallidium commented 3 years ago

added LN regularization in 23c671a

LumenPallidium commented 3 years ago

added KL divergence loss for the autoencoder in 6bc5afdc07ed84adffcf0b5fd1d7ef1c12ecdb7f