Arnaudding001 commited on
Commit
30131c2
1 Parent(s): 648ad35

Update encoder_psp.py

Browse files
Files changed (1) hide show
  1. encoder_psp.py +2 -2
encoder_psp.py CHANGED
@@ -7,8 +7,8 @@ import math
7
 
8
  import torch
9
  from torch import nn
10
- from model.encoder.encoders import psp_encoders
11
- from model.stylegan.model import Generator
12
 
13
  def get_keys(d, name):
14
  if 'state_dict' in d:
 
7
 
8
  import torch
9
  from torch import nn
10
+ from encoder_encoders import psp_encoders
11
+ from stylegan_model import Generator
12
 
13
  def get_keys(d, name):
14
  if 'state_dict' in d: