bugfix

utils.py

@@ -43,4 +43,5 @@ def load_default_gen(in_gpu=False, fname_pkl=None):
         G = pickle.load(f)['G_ema']  # torch.nn.ModuleDict
     if in_gpu:
         G = G.cuda()
+    return G