PuxAI commited on
Commit
80af6fe
·
verified ·
1 Parent(s): c212981

Logs step 10

Browse files
Files changed (2) hide show
  1. logs/train.log +103 -0
  2. logs/val.log +1 -0
logs/train.log ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 26-01-31 02:58:51.045 - INFO: name: DEBUG_TEST_RUN
2
+ phase: train
3
+ gpu_ids: [0]
4
+ path:[
5
+ log: experiments/DEBUG_TEST_RUN_260131_025851/logs
6
+ tb_logger: experiments/DEBUG_TEST_RUN_260131_025851/tb_logger
7
+ results: experiments/DEBUG_TEST_RUN_260131_025851/results
8
+ checkpoint: experiments/DEBUG_TEST_RUN_260131_025851/checkpoint
9
+ resume_state: None
10
+ experiments_root: experiments/DEBUG_TEST_RUN_260131_025851
11
+ ]
12
+ datasets:[
13
+ train:[
14
+ name: ICPR_Train
15
+ mode: ICPR
16
+ dataroot: ./dataset/train
17
+ width: 128
18
+ height: 64
19
+ batch_size: 2
20
+ num_workers: 0
21
+ use_shuffle: True
22
+ ]
23
+ val:[
24
+ name: ICPR_Val
25
+ mode: ICPR
26
+ dataroot: ./dataset/train
27
+ width: 128
28
+ height: 64
29
+ batch_size: 1
30
+ num_workers: 0
31
+ use_shuffle: False
32
+ data_len: 3
33
+ ]
34
+ ]
35
+ model:[
36
+ finetune_norm: False
37
+ unet:[
38
+ in_channel: 6
39
+ out_channel: 3
40
+ inner_channel: 64
41
+ channel_multiplier: [1, 2, 4, 8]
42
+ attn_res: [16]
43
+ res_blocks: 2
44
+ dropout: 0.2
45
+ ]
46
+ beta_schedule:[
47
+ train:[
48
+ schedule: linear
49
+ n_timestep: 1000
50
+ linear_start: 1e-06
51
+ linear_end: 0.01
52
+ ]
53
+ val:[
54
+ schedule: linear
55
+ n_timestep: 1000
56
+ linear_start: 1e-06
57
+ linear_end: 0.01
58
+ ]
59
+ ]
60
+ diffusion:[
61
+ image_size: 128
62
+ channels: 3
63
+ conditional: True
64
+ ]
65
+ ]
66
+ train:[
67
+ use_prerain_MTA: False
68
+ resume_training: False
69
+ MTA: ./best_377.pt
70
+ n_iter: 20
71
+ val_freq: 10
72
+ save_checkpoint_freq: 10
73
+ print_freq: 2
74
+ optimizer:[
75
+ type: adam
76
+ lr: 0.0001
77
+ ]
78
+ ]
79
+ wandb:[
80
+ project: ICPR-LP-Diff
81
+ ]
82
+ distributed: False
83
+ log_wandb_ckpt: False
84
+ log_eval: False
85
+ enable_wandb: False
86
+
87
+ 26-01-31 02:59:02.275 - INFO: Dataset [ICPRDataset - ICPR_Train] is created.
88
+ 26-01-31 02:59:07.513 - INFO: Dataset [ICPRDataset - ICPR_Val] is created.
89
+ 26-01-31 02:59:07.513 - INFO: Initial Dataset Finished
90
+ 26-01-31 02:59:08.877 - INFO: Initialization method [orthogonal]
91
+ 26-01-31 02:59:13.684 - INFO: Model [DDPM] is created.
92
+ 26-01-31 02:59:13.684 - INFO: Initial Model Finished
93
+ 26-01-31 02:59:27.083 - INFO: <epoch: 1, iter: 2> l_pix: 8.9346e-01
94
+ 26-01-31 02:59:27.440 - INFO: <epoch: 1, iter: 4> l_pix: 7.9007e-01
95
+ 26-01-31 02:59:27.682 - INFO: <epoch: 1, iter: 6> l_pix: 8.2409e-01
96
+ 26-01-31 02:59:27.992 - INFO: <epoch: 1, iter: 8> l_pix: 7.5086e-01
97
+ 26-01-31 02:59:28.262 - INFO: <epoch: 1, iter: 10> l_pix: 7.5090e-01
98
+ 26-01-31 03:00:06.459 - INFO: Saved model in [experiments/DEBUG_TEST_RUN_260131_025851/checkpoint/I10_E1_gen_best.pth] ...
99
+ 26-01-31 03:00:06.461 - INFO: # Validation # PSNR: 7.0245e+00
100
+ 26-01-31 03:00:06.461 - INFO: <epoch: 1, iter: 10> psnr: 7.0245e+00 loss: 9.6489e-01
101
+ 26-01-31 03:00:06.462 - INFO: Saving models and training states.
102
+ 26-01-31 03:00:08.018 - INFO: Saved model in [experiments/DEBUG_TEST_RUN_260131_025851/checkpoint/I10_E1_gen.pth] ...
103
+ 26-01-31 03:00:08.019 - INFO: ☁️ Syncing to Hugging Face: PuxAI/ICPR...
logs/val.log ADDED
@@ -0,0 +1 @@
 
 
1
+ 26-01-31 03:00:06.461 - INFO: <epoch: 1, iter: 10> psnr: 7.0245e+00 loss: 9.6489e-01