mingyuanzhou commited on Oct 27, 2023

Commit

bccc6c4

1 Parent(s): fbde75e

add checkpoints for plotting

Browse files

This view is limited to 50 files because it contains too many changes. See raw diff

Files changed (50) hide show

checkpoint_Beta_ELBO_B288/.DS_Store +0 -0
checkpoint_Beta_ELBO_B288/.ipynb_checkpoints/betadiff_x0_hat-checkpoint.png +0 -0
checkpoint_Beta_ELBO_B288/.ipynb_checkpoints/image90-checkpoint.png +0 -0
checkpoint_Beta_ELBO_B288/betadiff_x0_hat.png +0 -0
checkpoint_Beta_ELBO_B288/checkpoint.txt +3 -0
checkpoint_Beta_ELBO_B288/image27.png +0 -0
checkpoint_Beta_ELBO_B288/image27_1.png +0 -0
checkpoint_Beta_ELBO_B288/image29.png +0 -0
checkpoint_Beta_ELBO_B288/image29_1.png +0 -0
checkpoint_Beta_ELBO_B288/image30.png +0 -0
checkpoint_Beta_ELBO_B288/image30_1.png +0 -0
checkpoint_Beta_ELBO_B288/image54.png +0 -0
checkpoint_Beta_ELBO_B288/image54_1.png +0 -0
checkpoint_Beta_ELBO_B288/image74.png +0 -0
checkpoint_Beta_ELBO_B288/image74_1.png +0 -0
checkpoint_Beta_ELBO_B288/image80.png +0 -0
checkpoint_Beta_ELBO_B288/image80_1.png +0 -0
checkpoint_Beta_ELBO_B288/image9.png +0 -0
checkpoint_Beta_ELBO_B288/image90.png +0 -0
checkpoint_Beta_ELBO_B288/image90_1.png +0 -0
checkpoint_Beta_ELBO_B288/image93.png +0 -0
checkpoint_Beta_ELBO_B288/image93_1.png +0 -0
checkpoint_Beta_ELBO_B288/image9_1.png +0 -0
checkpoint_Beta_ELBO_B288/network-snapshot-200000.pkl +3 -0
checkpoint_Beta_ELBO_B288/training_options.json +77 -0
checkpoint_Beta_KLUB_B288/.DS_Store +0 -0
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/betadiff_x0_hat-checkpoint.png +0 -0
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image27-checkpoint.png +0 -0
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image74-checkpoint.png +0 -0
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image9-checkpoint.png +0 -0
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image90-checkpoint.png +0 -0
checkpoint_Beta_KLUB_B288/betadiff_x0_hat.png +0 -0
checkpoint_Beta_KLUB_B288/image27.png +0 -0
checkpoint_Beta_KLUB_B288/image27_1.png +0 -0
checkpoint_Beta_KLUB_B288/image29.png +0 -0
checkpoint_Beta_KLUB_B288/image29_1.png +0 -0
checkpoint_Beta_KLUB_B288/image30.png +0 -0
checkpoint_Beta_KLUB_B288/image30_1.png +0 -0
checkpoint_Beta_KLUB_B288/image54.png +0 -0
checkpoint_Beta_KLUB_B288/image54_1.png +0 -0
checkpoint_Beta_KLUB_B288/image74.png +0 -0
checkpoint_Beta_KLUB_B288/image74_1.png +0 -0
checkpoint_Beta_KLUB_B288/image80.png +0 -0
checkpoint_Beta_KLUB_B288/image80_1.png +0 -0
checkpoint_Beta_KLUB_B288/image9.png +0 -0
checkpoint_Beta_KLUB_B288/image90.png +0 -0
checkpoint_Beta_KLUB_B288/image90_1.png +0 -0
checkpoint_Beta_KLUB_B288/image93.png +0 -0
checkpoint_Beta_KLUB_B288/image93_1.png +0 -0
checkpoint_Beta_KLUB_B288/image9_1.png +0 -0

checkpoint_Beta_ELBO_B288/.DS_Store ADDED Viewed

Binary file (6.15 kB). View file

checkpoint_Beta_ELBO_B288/.ipynb_checkpoints/betadiff_x0_hat-checkpoint.png ADDED Viewed

checkpoint_Beta_ELBO_B288/.ipynb_checkpoints/image90-checkpoint.png ADDED Viewed

checkpoint_Beta_ELBO_B288/betadiff_x0_hat.png ADDED Viewed

checkpoint_Beta_ELBO_B288/checkpoint.txt ADDED Viewed

	@@ -0,0 +1,3 @@


1	+ Link to checkpoint:
2	+
3	+ https://drive.google.com/file/d/1jaMaDl-R1IVFlPlrj9nd9dNZWIutUU1s/view?usp=drive_link

checkpoint_Beta_ELBO_B288/image27.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image27_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image29.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image29_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image30.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image30_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image54.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image54_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image74.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image74_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image80.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image80_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image9.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image90.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image90_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image93.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image93_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/image9_1.png ADDED Viewed

checkpoint_Beta_ELBO_B288/network-snapshot-200000.pkl ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:f85ac2428caa93be7bd7ef7e280a658a1379cd0869f2bd36614f2cbca2578d0e
+size 223199962

checkpoint_Beta_ELBO_B288/training_options.json ADDED Viewed

	@@ -0,0 +1,77 @@

+{
+  "dataset_kwargs": {
+    "class_name": "training.dataset.ImageFolderDataset",
+    "path": "datasets/cifar10-32x32.zip",
+    "use_labels": false,
+    "xflip": false,
+    "cache": true,
+    "resolution": 32,
+    "max_size": 50000
+  },
+  "data_loader_kwargs": {
+    "pin_memory": true,
+    "num_workers": 1,
+    "prefetch_factor": 2
+  },
+  "network_kwargs": {
+    "model_type": "SongUNet",
+    "embedding_type": "positional",
+    "encoder_type": "standard",
+    "decoder_type": "standard",
+    "channel_mult_noise": 1,
+    "resample_filter": [
+      1,
+      1
+    ],
+    "model_channels": 128,
+    "channel_mult": [
+      2,
+      2,
+      2
+    ],
+    "class_name": "training.networks.BetaDiffPrecond",
+    "augment_dim": 9,
+    "dropout": 0.13,
+    "use_fp16": false
+  },
+  "loss_kwargs": {
+    "class_name": "training.loss.BetaDiffLoss",
+    "eta": 10000.0,
+    "sigmoid_start": 10.0,
+    "sigmoid_end": -13.0,
+    "sigmoid_power": 1.0,
+    "Scale": 0.39,
+    "Shift": 0.6,
+    "lossType": "KLUB_AS"
+  },
+  "optimizer_kwargs": {
+    "class_name": "torch.optim.Adam",
+    "lr": 0.0002,
+    "betas": [
+      0.9,
+      0.999
+    ],
+    "eps": 1e-08
+  },
+  "augment_kwargs": {
+    "class_name": "training.augment.AugmentPipe",
+    "p": 0.12,
+    "xflip": 100000000.0,
+    "yflip": 1,
+    "scale": 1,
+    "rotate_frac": 1,
+    "aniso": 1,
+    "translate_frac": 1
+  },
+  "total_kimg": 200000,
+  "ema_halflife_kimg": 500,
+  "batch_size": 288,
+  "batch_gpu": null,
+  "loss_scaling": 1.0,
+  "cudnn_benchmark": true,
+  "kimg_per_tick": 50,
+  "snapshot_ticks": 50,
+  "state_dump_ticks": 500,
+  "seed": 17439077,
+  "run_dir": "betadiff-train-runs/00018-cifar10-32x32-uncond-ddpmpp-betadiff-gpus8-batch288-fp32"
+}