new

Code/train.py

@@ -110,7 +110,7 @@ def train(model, name, seed,gradient_accumulation_steps,mixup, threshold, lambda
 					#add mira 1 line
 					#new_labels_batch = new_labels_batch.to(torch.float64)
 
-					loss_2=loss_fct(logits.view(-1, 2).to("cuda"), new_labels_batch.view(-1).to("cuda").to(torch.float))
+					loss_2=loss_fct(logits.view(-1, 2).to("cuda"), new_labels_batch.view(-1).to("cuda").to(torch.float32))
 					print("MixUp Loss: ", loss_2)
 					#update entire model
 					loss_2.backward()