Skip to content

Commit 266a96c

Browse files
author
zhedong
committed
change gamma to 8
1 parent 5121808 commit 266a96c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def train_model(model, criterion, optimizer, scheduler, num_epochs=25):
216216
if opt.contrast:
217217
criterion_contrast = losses.ContrastiveLoss(pos_margin=0, neg_margin=1)
218218
if opt.instance:
219-
criterion_instance = InstanceLoss(gamma=1)
219+
criterion_instance = InstanceLoss(gamma=8)
220220
if opt.sphere:
221221
criterion_sphere = losses.SphereFaceLoss(num_classes=opt.nclasses, embedding_size=512, margin=4)
222222
for epoch in range(num_epochs):

0 commit comments

Comments
 (0)