Python neko_cos_loss示例

示例#1

文件： HXOS.py 项目： lancercat/OSOCR

 def setuploss(this):
     this.criterion_CE = nn.CrossEntropyLoss().cuda()
     this.url = neko_unknown_ranking_loss()
     this.cosloss = neko_cos_loss().cuda()
     this.wcls = 1
     this.wsim = 0
     this.wmar = 0

示例#2

文件： HXOS.py 项目： lancercat/OSOCR

 def setuploss(this):
     this.criterion_CE = nn.CrossEntropyLoss().cuda()
     this.cosloss = neko_cos_loss().cuda()