tristan-deep commited on
Commit
5f96c6b
·
verified ·
1 Parent(s): f67a01c

Upload 4 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ training_checkpoints/ckpt-7.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
config.yaml ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ activation: relu
2
+ batch_size: 16
3
+ beta_max: 50.0
4
+ beta_min: 0.1
5
+ channels: 64
6
+ color_mode: rgb
7
+ corrector: none
8
+ dataset_name: mnist
9
+ drop_prob: null
10
+ embed_dim: 256
11
+ epochs: 100
12
+ eval_freq: 1
13
+ image_range:
14
+ - 0
15
+ - 1
16
+ image_shape:
17
+ - 64
18
+ - 64
19
+ - 3
20
+ image_size: 64
21
+ kernel_size: 3
22
+ likelihood_weighting: false
23
+ lr: 0.0002
24
+ model_name: score
25
+ n_steps_each: 1
26
+ noise_removal: true
27
+ normalization: batch
28
+ num_plot_img: 16
29
+ num_scales: 1000
30
+ patch_size: null
31
+ predictor: euler_maruyama
32
+ probability_flow: false
33
+ reduce_mean: false
34
+ sampling_method: pc
35
+ save_freq: 10
36
+ score_backbone: NCSNv2
37
+ sde: simple
38
+ seed: 1234
39
+ sigma: 25.0
40
+ sigma_max: 90.0
41
+ sigma_min: 0.01
42
+ snr: 0.17
43
+ upmode: upconv
training_checkpoints/checkpoint ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ model_checkpoint_path: "ckpt-7"
2
+ all_model_checkpoint_paths: "ckpt-7"
training_checkpoints/ckpt-7.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad62569d9f9a8e3c2cbc26faa6a89cf10139cb62c7d3cbe9e4cdf75bde8ee5f8
3
+ size 106891468
training_checkpoints/ckpt-7.index ADDED
Binary file (31.3 kB). View file