Johannes Kolbe
commited on
Commit
•
7d0a876
1
Parent(s):
8c56b3d
animegan checkpoint added
Browse files- .gitignore +0 -2
- checkpoints/stylegan_animeface512.pth +3 -0
- utils.py +1 -1
.gitignore
CHANGED
|
@@ -5,7 +5,6 @@ __pycache__/
|
|
| 5 |
/.idea/
|
| 6 |
*.sw[pon]
|
| 7 |
|
| 8 |
-
/checkpoints/
|
| 9 |
/results/
|
| 10 |
*.jpg
|
| 11 |
*.png
|
|
@@ -22,7 +21,6 @@ __pycache__/
|
|
| 22 |
*.zip
|
| 23 |
events.*
|
| 24 |
|
| 25 |
-
*.pth
|
| 26 |
*.pkl
|
| 27 |
*.h5
|
| 28 |
*.dat
|
|
|
|
| 5 |
/.idea/
|
| 6 |
*.sw[pon]
|
| 7 |
|
|
|
|
| 8 |
/results/
|
| 9 |
*.jpg
|
| 10 |
*.png
|
|
|
|
| 21 |
*.zip
|
| 22 |
events.*
|
| 23 |
|
|
|
|
| 24 |
*.pkl
|
| 25 |
*.h5
|
| 26 |
*.dat
|
checkpoints/stylegan_animeface512.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5ce84e28171aa6d4a4b2984a4ae67a3663da98700dda9137d20662abc9356990
|
| 3 |
+
size 307644761
|
utils.py
CHANGED
|
@@ -87,7 +87,7 @@ def load_generator(model_name):
|
|
| 87 |
generator.load_state_dict(checkpoint['generator_smooth'])
|
| 88 |
else:
|
| 89 |
generator.load_state_dict(checkpoint['generator'])
|
| 90 |
-
generator = generator.cuda()
|
| 91 |
generator.eval()
|
| 92 |
print(f'Finish loading checkpoint.')
|
| 93 |
return generator
|
|
|
|
| 87 |
generator.load_state_dict(checkpoint['generator_smooth'])
|
| 88 |
else:
|
| 89 |
generator.load_state_dict(checkpoint['generator'])
|
| 90 |
+
#generator = generator.cuda()
|
| 91 |
generator.eval()
|
| 92 |
print(f'Finish loading checkpoint.')
|
| 93 |
return generator
|