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

change final poolingfunction to control h,w size (instead of defaulting to 1,1) #17

Closed LumenPallidium closed 3 years ago

LumenPallidium commented 3 years ago

done in 5683fc7