BecomeAllan
/

ML-SLRC

Model card Files Files and versions Community

Allan Victor commited on May 26, 2022

Commit

d8795c2

•

1 Parent(s): fac590f

Update Util_funs.py

Files changed (1) hide show

Util_funs.py +1 -1

Util_funs.py CHANGED Viewed

@@ -242,7 +242,7 @@ def train_loop(data_train_loader, data_test_loader, model, device, epoch = 4, lr
         input_ids, attention_mask,q_token_type_ids, label_id = batch
         # Predictions
-         _, feature, _ = model_meta(input_ids, attention_mask,q_token_type_ids, labels = label_id.squeeze())
  #       prediction = prediction.detach().cpu().squeeze()
   #      label_id = label_id.detach().cpu()

         input_ids, attention_mask,q_token_type_ids, label_id = batch
         # Predictions
+        _, feature, _ = model_meta(input_ids, attention_mask,q_token_type_ids, labels = label_id.squeeze())
  #       prediction = prediction.detach().cpu().squeeze()
   #      label_id = label_id.detach().cpu()