Skip to content

Latest commit

 

History

History
14 lines (9 loc) · 463 Bytes

README.md

File metadata and controls

14 lines (9 loc) · 463 Bytes

scs_pytorch

Pytorch implementation of the Sharpened Cosine Similarity layer mentioned in this tweet.

The current version supports square kernels with arbitrary kernel sizes.

Part of the implementation borrows from dvisockas/cos_sim.

To do

  • depth-wise conv (with groups arguement)
  • non-square kernels
  • efficiency
  • demo training code