Skip to content

v0.1.31

Compare
Choose a tag to compare
@monatis monatis released this 06 Oct 20:42
· 13 commits to master since this release
f2be2a4

What's new

  • Add semi-hard mining to Triplet Loss, and introduce Cross-Batch Memory support(#175).
  • Add soft margin variant to Hard Triplet Loss (#178).
  • Support multiple EncoderHead in a SwitchHead, most useful for two-tower encoders like CLIP, and enable to pass meta data from Encoder to EncoderHead, also useful for passing masks, e.g., attention masks (#180).