Upload 8 files

checkpoints/uqdm-big/checkpoint.pt

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

checkpoints/uqdm-big/config.json

@@ -0,0 +1,45 @@
+{
+  "restore_ckpt": "checkpoint.pt",
+  "model": {
+    "n_timesteps": 4,
+    "prior_type": "logistic",
+    "learned_prior_scale": true,
+    "base_prior_scale": "forward_kernel",
+    "noise_schedule": "learned_linear",
+    "fix_gamma_max": true,
+    "gamma_min": -12.3,
+    "gamma_max": 6.0,
+    "attention_everywhere": false,
+    "use_fourier_features": true,
+    "ema_rate": 0.9999,
+    "n_attention_heads": 1,
+    "n_channels": 3,
+    "vocab_size": 256,
+    "embedding_dim": 256,
+    "n_blocks": 64,
+    "norm_groups": 32,
+    "dropout_prob": 0.0
+  },
+  "training": {
+    "n_steps": 500000,
+    "log_metrics_every_steps": 1000,
+    "checkpoint_every_steps": 10000,
+    "eval_every_steps": 10000,
+    "eval_steps_to_run": 50
+  },
+  "optim": {
+    "weight_decay": 0,
+    "beta1": 0.9,
+    "eps": 1e-08,
+    "warmup": 1000,
+    "lr": 0.0002,
+    "grad_clip_norm": 1.0
+  },
+  "data": {
+    "shuffle": false,
+    "pin_memory": true,
+    "batch_size": 2,
+    "num_workers": 1,
+    "data_spec": "imagenet"
+  }
+}

checkpoints/uqdm-medium/checkpoint.pt

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

checkpoints/uqdm-medium/config.json

@@ -0,0 +1,45 @@
+{
+  "restore_ckpt": "checkpoint.pt",
+  "model": {
+    "n_timesteps": 4,
+    "prior_type": "logistic",
+    "learned_prior_scale": true,
+    "base_prior_scale": "forward_kernel",
+    "noise_schedule": "learned_linear",
+    "fix_gamma_max": true,
+    "gamma_min": -12.3,
+    "gamma_max": 6.0,
+    "attention_everywhere": false,
+    "use_fourier_features": true,
+    "ema_rate": 0.9999,
+    "n_attention_heads": 1,
+    "n_channels": 3,
+    "vocab_size": 256,
+    "embedding_dim": 256,
+    "n_blocks": 32,
+    "norm_groups": 32,
+    "dropout_prob": 0.0
+  },
+  "training": {
+    "n_steps": 500000,
+    "log_metrics_every_steps": 1000,
+    "checkpoint_every_steps": 10000,
+    "eval_every_steps": 10000,
+    "eval_steps_to_run": 50
+  },
+  "optim": {
+    "weight_decay": 0,
+    "beta1": 0.9,
+    "eps": 1e-08,
+    "warmup": 1000,
+    "lr": 0.0002,
+    "grad_clip_norm": 1.0
+  },
+  "data": {
+    "shuffle": false,
+    "pin_memory": true,
+    "batch_size": 2,
+    "num_workers": 1,
+    "data_spec": "imagenet"
+  }
+}

checkpoints/uqdm-small/checkpoint.pt

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

checkpoints/uqdm-small/config.json

@@ -0,0 +1,45 @@
+{
+  "restore_ckpt": "checkpoint.pt",
+  "model": {
+    "n_timesteps": 4,
+    "prior_type": "logistic",
+    "learned_prior_scale": true,
+    "base_prior_scale": "forward_kernel",
+    "noise_schedule": "learned_linear",
+    "fix_gamma_max": true,
+    "gamma_min": -12.3,
+    "gamma_max": 6.0,
+    "attention_everywhere": false,
+    "use_fourier_features": true,
+    "ema_rate": 0.9999,
+    "n_attention_heads": 1,
+    "n_channels": 3,
+    "vocab_size": 256,
+    "embedding_dim": 64,
+    "n_blocks": 8,
+    "norm_groups": 16,
+    "dropout_prob": 0.0
+  },
+  "training": {
+    "n_steps": 500000,
+    "log_metrics_every_steps": 1000,
+    "checkpoint_every_steps": 10000,
+    "eval_every_steps": 10000,
+    "eval_steps_to_run": 50
+  },
+  "optim": {
+    "weight_decay": 0,
+    "beta1": 0.9,
+    "eps": 1e-08,
+    "warmup": 1000,
+    "lr": 0.0002,
+    "grad_clip_norm": 1.0
+  },
+  "data": {
+    "shuffle": false,
+    "pin_memory": true,
+    "batch_size": 2,
+    "num_workers": 1,
+    "data_spec": "imagenet"
+  }
+}

checkpoints/uqdm-tiny/checkpoint.pt

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

checkpoints/uqdm-tiny/config.json

@@ -0,0 +1,45 @@
+{
+  "restore_ckpt": "checkpoint.pt",
+  "model": {
+    "n_timesteps": 4,
+    "prior_type": "logistic",
+    "learned_prior_scale": true,
+    "base_prior_scale": "forward_kernel",
+    "noise_schedule": "learned_linear",
+    "fix_gamma_max": true,
+    "gamma_min": -12.3,
+    "gamma_max": 6.0,
+    "attention_everywhere": false,
+    "use_fourier_features": true,
+    "ema_rate": 0.9999,
+    "n_attention_heads": 1,
+    "n_channels": 3,
+    "vocab_size": 256,
+    "embedding_dim": 32,
+    "n_blocks": 1,
+    "norm_groups": 8,
+    "dropout_prob": 0.0
+  },
+  "training": {
+    "n_steps": 500000,
+    "log_metrics_every_steps": 1000,
+    "checkpoint_every_steps": 10000,
+    "eval_every_steps": 10000,
+    "eval_steps_to_run": 50
+  },
+  "optim": {
+    "weight_decay": 0,
+    "beta1": 0.9,
+    "eps": 1e-08,
+    "warmup": 1000,
+    "lr": 0.0002,
+    "grad_clip_norm": 1.0
+  },
+  "data": {
+    "shuffle": false,
+    "pin_memory": true,
+    "batch_size": 2,
+    "num_workers": 1,
+    "data_spec": "imagenet"
+  }
+}