Logs step 10
Browse files- logs/train.log +103 -0
- 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
|