Skip to content

PyTorch implementation of 2D Sharpened Cosine Similarity layer

Notifications You must be signed in to change notification settings

ZeWang95/scs_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 

Repository files navigation

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

About

PyTorch implementation of 2D Sharpened Cosine Similarity layer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages