nousr commited on
Commit
9712020
1 Parent(s): 40aea42

Upload prior/prior_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. prior/prior_config.json +8 -6
prior/prior_config.json CHANGED
@@ -34,15 +34,15 @@
34
  "condition_on_text_encodings": true
35
  },
36
  "data": {
37
- "batch_size": 220,
38
  "num_data_points": 500000000,
39
- "eval_every_seconds": 1800,
40
  "image_url": "s3://s-laion5b/embeddings/vit-l-14/laion2B-en/img_emb/",
41
  "meta_url": "s3://s-laion5b/embeddings/vit-l-14/laion2B-en/metadata/",
42
  "splits": {
43
  "train": 0.9,
44
- "val": 6e-5,
45
- "test":0.09994000000000003
46
  }
47
  },
48
  "train": {
@@ -66,11 +66,13 @@
66
  "log_type": "wandb",
67
  "wandb_entity": "nousr_laion",
68
  "wandb_project": "dalle2_diffusion_prior",
69
- "wandb_resume": false,
 
70
  "verbose": true
71
  },
72
  "load": {
73
- "resume": false
 
74
  },
75
  "save": [
76
  {
 
34
  "condition_on_text_encodings": true
35
  },
36
  "data": {
37
+ "batch_size": 225,
38
  "num_data_points": 500000000,
39
+ "eval_every_seconds": 1600,
40
  "image_url": "s3://s-laion5b/embeddings/vit-l-14/laion2B-en/img_emb/",
41
  "meta_url": "s3://s-laion5b/embeddings/vit-l-14/laion2B-en/metadata/",
42
  "splits": {
43
  "train": 0.9,
44
+ "val": 1.2e-5,
45
+ "test": 0.09998799999999998
46
  }
47
  },
48
  "train": {
 
66
  "log_type": "wandb",
67
  "wandb_entity": "nousr_laion",
68
  "wandb_project": "dalle2_diffusion_prior",
69
+ "wandb_resume": true,
70
+ "wandb_run_id": "20jdn8i2",
71
  "verbose": true
72
  },
73
  "load": {
74
+ "load_from": "local",
75
+ "file_path": "/fsx/nousr/.prior/latest_checkpoint.pth"
76
  },
77
  "save": [
78
  {