Add files via upload

ericlearning · Dec 22, 2019 · 93fcc9d · 93fcc9d
1 parent 583086d
commit 93fcc9d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/trainers/trainer.py b/trainers/trainer.py
@@ -15,7 +15,7 @@
 
 class Trainer():
  def __init__(self, loss_type, netD, netG, device, train_ds, lr_D = 0.0002, lr_G = 0.0002, resample = True, weight_clip = None, use_gradient_penalty = False, drift = 0.001, loss_interval = 50, image_interval = 50, save_img_dir = 'saved_images/'):
- self.loss_type = loss_type
+ self.loss_type, self.device = loss_type, device
  self.require_type = get_require_type(self.loss_type)
  self.loss = get_gan_loss(self.device, self.loss_type)
 

diff --git a/utils.py b/utils.py
@@ -12,6 +12,7 @@
 from sklearn.metrics import confusion_matrix
 from scipy.io import wavfile
 from PIL import Image
+from losses.losses import *
 
 def set_lr(optimizer, lrs):
  if(len(lrs) == 1):