Update train_coarse.py

Siyun-Jung · Jul 28, 2023 · 4a665fa · 4a665fa
1 parent deb729d
commit 4a665fa
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/train_coarse.py b/train_coarse.py
@@ -71,8 +71,6 @@ def train_generator(input, Mask, Maskmimi, label, traing_loss, iteration):
  # Coarse generator
  coarse_pred = gen_coarse(input).to(device)
 
-
-
  # Loss calculations and backpropagation
  finalg_loss = calculate_gen_loss(coarse_pred, Mask, Maskmimi, label)
  traing_loss += finalg_loss.item()
@@ -103,4 +101,4 @@ def calculate_gen_loss(fine_pred, Mask, Maskmimi, label):
  return finalg_loss
 
 # Call the main function
-train_model_coarse(num_epoch)
+train_model_coarse(num_epoch)