Hugging Face's logo

Spaces:
Legola
/
StyleGAN
Build error

App Files Community
Legola commited on
Commit
19e21cf
·
1 Parent(s): 666de43

Update model.py

Browse files

Added torch.nn as nn

Files changed (1) hide show
  1. model.py +2 -0
model.py CHANGED
@@ -1,4 +1,6 @@
1
 
 
 
2
  #Content Loss
3
  class ContentLoss(nn.Module):
4
 
 
1
 
2
+ import torch
3
+ import torch.nn as nn
4
  #Content Loss
5
  class ContentLoss(nn.Module):
6