Merge pull request #121 from hyperion-ml/tyche-cleanup

jesus-villalba · web-flow · commit c4c9eee0acab · 2022-03-25T12:28:08.000-04:00
fix cos-scale
diff --git a/hyperion/torch/models/xvectors/xvector.py b/hyperion/torch/models/xvectors/xvector.py
@@ -501,7 +501,7 @@ def rebuild_output_layer(
             # if we change the number of classes or the loss-type
             # we need to reinitiate the last layer
             self.classif_net.rebuild_output_layer(
-                num_classes, loss_type, s, margin, margin_warmup_epochs
+                num_classes, loss_type, cos_scale, margin, margin_warmup_epochs
             )
             return
 

Original file line number	Diff line number	Diff line change
`@@ -501,7 +501,7 @@ def rebuild_output_layer(`
`501`	`501`	`# if we change the number of classes or the loss-type`
`502`	`502`	`# we need to reinitiate the last layer`
`503`	`503`	`self.classif_net.rebuild_output_layer(`
`504`		`- num_classes, loss_type, s, margin, margin_warmup_epochs`
	`504`	`+ num_classes, loss_type, cos_scale, margin, margin_warmup_epochs`
`505`	`505`	`)`
`506`	`506`	`return`
`507`	`507`