Skip to content

release v0.0.1

Latest
Compare
Choose a tag to compare
@yakhyo yakhyo released this 14 Mar 13:15
· 10 commits to main since this release

These models trained with 10 epochs. Model weights saved in f16 format (half tensor).

Model names:

  • ce.pt - model trained with cross entropy loss function
  • dice.pt - model trained with dice loss function
  • dicece.pt - model trained with dice cross entropy loss function
  • focal.pt- model trained with focal loss function