add rsim to train.py

kellymarchisio · Feb 27, 2023 · 949bb62 · 949bb62
1 parent 27c45f4
commit 949bb62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/train.py b/src/train.py
@@ -515,7 +515,7 @@ def train(args, device, vocab, train_file, total_train_n, total_lines,
                     iso_loss_unscaled = torch.linalg.norm(model_vecs_tmp - loaded_vecs)
                     iso_loss = iso_loss_unscaled / len(loaded_vecs)
                 elif args.loss == 'rs':
-                    iso_loss = gh.diffble_rs_distance(model_vecs_tmp,
+                    iso_loss = iso.diffble_rs_distance(model_vecs_tmp,
                             loaded_vecs, device, args.mode == 'unsupervised')
                 elif args.loss == 'evs':
                     iso_loss = iso.diffble_evs_distance(model_vecs_tmp,