Upload 4 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+training_checkpoints/ckpt-8.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

config.yaml

@@ -0,0 +1,69 @@
+activation: relu
+batch_size: 16
+beta_max: 50.0
+beta_min: 0.1
+channels: 64
+color_mode: rgb
+corrector: none
+d_apply_batchnorm: false
+d_channels:
+- 64
+- 128
+- 256
+- 256
+d_drop_prob: 0.3
+d_kernel_size: 5
+d_lr: 0.0001
+d_output_activation: null
+d_steps: 3
+dataset_name: celeba
+drop_prob: null
+embed_dim: 256
+epochs: 100
+eval_freq: 1
+g_apply_batchnorm: true
+g_channels:
+- 256
+- 256
+- 256
+- 128
+g_drop_prob: null
+g_first_dense_size: 128
+g_kernel_size: 5
+g_lr: 0.0001
+g_output_activation: tanh
+g_upmode: upconv
+gp_weight: 1
+image_range:
+- 0
+- 1
+image_shape:
+- 64
+- 64
+- 3
+image_size: 64
+kernel_size: 3
+label_sigma: 0.05
+latent_dim: 64
+likelihood_weighting: false
+lr: 0.0002
+model_name: score
+n_steps_each: 1
+noise_removal: true
+normalization: batch
+num_plot_img: 16
+num_scales: 1000
+patch_size: null
+predictor: euler_maruyama
+probability_flow: false
+reduce_mean: false
+sampling_method: pc
+save_freq: 10
+score_backbone: NCSNv2
+sde: simple
+seed: 1234
+sigma: 25.0
+sigma_max: 90.0
+sigma_min: 0.01
+snr: 0.17
+upmode: upconv

training_checkpoints/checkpoint

@@ -0,0 +1,2 @@
+model_checkpoint_path: "ckpt-8"
+all_model_checkpoint_paths: "ckpt-8"

training_checkpoints/ckpt-8.data-00000-of-00001

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