WGAN-GP Improvements - Githubissues

wvabrinskas / Neuron

A neural network library for Swift

MIT License

104 stars 6 forks source link

Closed wvabrinskas closed 8 months ago

wvabrinskas commented 8 months ago

MSE Loss Function updates
WGAN-GP gradient fixes
Tensor mean function
Sped up Convolution by around ~25 - 50% by allowing them to perform multithreaded operations