Upload eigengan_ffhq checkpoints and training logs
Browse files- eigengan_ffhq/ckpt/step399999/ema.pt +3 -0
- eigengan_ffhq/ckpt/step399999/ema_model.pt +3 -0
- eigengan_ffhq/ckpt/step399999/meta.pt +3 -0
- eigengan_ffhq/ckpt/step399999/model.pt +3 -0
- eigengan_ffhq/ckpt/step399999/optimizer.pt +3 -0
- eigengan_ffhq/config-2024-01-03-12-10-42.yaml +68 -0
- eigengan_ffhq/output-2024-01-03-12-10-42.log +0 -0
- eigengan_ffhq/samples.zip +3 -0
- eigengan_ffhq/tensorboard/events.out.tfevents.1704255043.amax.596956.0 +3 -0
eigengan_ffhq/ckpt/step399999/ema.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:be34a266f197bee8c81ef0fdc1b463f74af21faebf1029e0ca027d07326b3460
|
3 |
+
size 91994867
|
eigengan_ffhq/ckpt/step399999/ema_model.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e50d03af702fd6210a373ba630370a56ac46d3e101e3ee92655d1d8c4081697b
|
3 |
+
size 146558789
|
eigengan_ffhq/ckpt/step399999/meta.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9ba18ce910f1d483b7c4f04b0d734899ae20c22aad14a06474d7c2015010127b
|
3 |
+
size 425
|
eigengan_ffhq/ckpt/step399999/model.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f71750153191cda451a8ace752318b9deb6ece355928bf1941695025066766e7
|
3 |
+
size 146552357
|
eigengan_ffhq/ckpt/step399999/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d8318c82b3b19de798c24b9dc29ecce90c6babd803e9c01a01351260357228ba
|
3 |
+
size 293129925
|
eigengan_ffhq/config-2024-01-03-12-10-42.yaml
ADDED
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 5678
|
2 |
+
data:
|
3 |
+
name: FFHQ
|
4 |
+
dataroot: ~/data/FFHQ/
|
5 |
+
img_size: 256
|
6 |
+
img_channels: 3
|
7 |
+
dataloader:
|
8 |
+
num_workers: 64
|
9 |
+
pin_memory: true
|
10 |
+
prefetch_factor: 2
|
11 |
+
G:
|
12 |
+
target: models.eigengan.Generator
|
13 |
+
params:
|
14 |
+
n_basis: 6
|
15 |
+
noise_dim: 512
|
16 |
+
out_dim: 3
|
17 |
+
dim: 16
|
18 |
+
dim_mults:
|
19 |
+
- 32
|
20 |
+
- 16
|
21 |
+
- 8
|
22 |
+
- 4
|
23 |
+
- 2
|
24 |
+
- 1
|
25 |
+
with_bn: false
|
26 |
+
with_tanh: true
|
27 |
+
D:
|
28 |
+
target: models.eigengan.Discriminator
|
29 |
+
params:
|
30 |
+
in_dim: 3
|
31 |
+
dim: 16
|
32 |
+
dim_mults:
|
33 |
+
- 2
|
34 |
+
- 4
|
35 |
+
- 8
|
36 |
+
- 16
|
37 |
+
- 32
|
38 |
+
- 32
|
39 |
+
with_bn: false
|
40 |
+
train:
|
41 |
+
n_steps: 500000
|
42 |
+
batch_size: 64
|
43 |
+
resume: null
|
44 |
+
print_freq: 100
|
45 |
+
save_freq: 10000
|
46 |
+
sample_freq: 1000
|
47 |
+
n_samples: 25
|
48 |
+
d_iters: 1
|
49 |
+
ortho_reg: 1.0
|
50 |
+
ema_decay: 0.999
|
51 |
+
loss_fn:
|
52 |
+
target: losses.hinge_loss.HingeLoss
|
53 |
+
params:
|
54 |
+
lambda_r1_reg: 10.0
|
55 |
+
optim_G:
|
56 |
+
target: torch.optim.Adam
|
57 |
+
params:
|
58 |
+
lr: 0.0001
|
59 |
+
betas:
|
60 |
+
- 0.5
|
61 |
+
- 0.999
|
62 |
+
optim_D:
|
63 |
+
target: torch.optim.Adam
|
64 |
+
params:
|
65 |
+
lr: 0.0001
|
66 |
+
betas:
|
67 |
+
- 0.5
|
68 |
+
- 0.999
|
eigengan_ffhq/output-2024-01-03-12-10-42.log
ADDED
The diff for this file is too large to render.
See raw diff
|
|
eigengan_ffhq/samples.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:99ce48d80ab26fad256ea7719540cf7737a1c59b8c68f24d97541cac013e08dd
|
3 |
+
size 1258726995
|
eigengan_ffhq/tensorboard/events.out.tfevents.1704255043.amax.596956.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f005e9f7cba3d9dc2f0f3611800b33fe07578573c3a6814a539ca1aaeee50ba3
|
3 |
+
size 129417518
|