Upload eigengan_ffhq checkpoints and training logs

eigengan_ffhq/ckpt/step399999/ema.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:be34a266f197bee8c81ef0fdc1b463f74af21faebf1029e0ca027d07326b3460
+size 91994867

eigengan_ffhq/ckpt/step399999/ema_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e50d03af702fd6210a373ba630370a56ac46d3e101e3ee92655d1d8c4081697b
+size 146558789

eigengan_ffhq/ckpt/step399999/meta.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ba18ce910f1d483b7c4f04b0d734899ae20c22aad14a06474d7c2015010127b
+size 425

eigengan_ffhq/ckpt/step399999/model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f71750153191cda451a8ace752318b9deb6ece355928bf1941695025066766e7
+size 146552357

eigengan_ffhq/ckpt/step399999/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d8318c82b3b19de798c24b9dc29ecce90c6babd803e9c01a01351260357228ba
+size 293129925

eigengan_ffhq/config-2024-01-03-12-10-42.yaml

@@ -0,0 +1,68 @@
+seed: 5678
+data:
+  name: FFHQ
+  dataroot: ~/data/FFHQ/
+  img_size: 256
+  img_channels: 3
+dataloader:
+  num_workers: 64
+  pin_memory: true
+  prefetch_factor: 2
+G:
+  target: models.eigengan.Generator
+  params:
+    n_basis: 6
+    noise_dim: 512
+    out_dim: 3
+    dim: 16
+    dim_mults:
+    - 32
+    - 16
+    - 8
+    - 4
+    - 2
+    - 1
+    with_bn: false
+    with_tanh: true
+D:
+  target: models.eigengan.Discriminator
+  params:
+    in_dim: 3
+    dim: 16
+    dim_mults:
+    - 2
+    - 4
+    - 8
+    - 16
+    - 32
+    - 32
+    with_bn: false
+train:
+  n_steps: 500000
+  batch_size: 64
+  resume: null
+  print_freq: 100
+  save_freq: 10000
+  sample_freq: 1000
+  n_samples: 25
+  d_iters: 1
+  ortho_reg: 1.0
+  ema_decay: 0.999
+  loss_fn:
+    target: losses.hinge_loss.HingeLoss
+    params:
+      lambda_r1_reg: 10.0
+  optim_G:
+    target: torch.optim.Adam
+    params:
+      lr: 0.0001
+      betas:
+      - 0.5
+      - 0.999
+  optim_D:
+    target: torch.optim.Adam
+    params:
+      lr: 0.0001
+      betas:
+      - 0.5
+      - 0.999

eigengan_ffhq/samples.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:99ce48d80ab26fad256ea7719540cf7737a1c59b8c68f24d97541cac013e08dd
+size 1258726995

eigengan_ffhq/tensorboard/events.out.tfevents.1704255043.amax.596956.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f005e9f7cba3d9dc2f0f3611800b33fe07578573c3a6814a539ca1aaeee50ba3
+size 129417518