예제 #1
0
파일: 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
0
파일: HXOS.py 프로젝트: lancercat/OSOCR
 def setuploss(this):
     this.criterion_CE = nn.CrossEntropyLoss().cuda()
     this.cosloss = neko_cos_loss().cuda()