diff --git a/checkpoints/ROMA_UNSB_001/loss_log.txt b/checkpoints/ROMA_UNSB_001/loss_log.txt index c7fad7f..bb13a7d 100644 --- a/checkpoints/ROMA_UNSB_001/loss_log.txt +++ b/checkpoints/ROMA_UNSB_001/loss_log.txt @@ -69,3 +69,15 @@ ================ Training Loss (Sun Feb 23 23:13:59 2025) ================ ================ Training Loss (Sun Feb 23 23:14:59 2025) ================ ================ Training Loss (Mon Feb 24 22:59:41 2025) ================ +================ Training Loss (Mon Feb 24 23:01:03 2025) ================ +================ Training Loss (Mon Feb 24 23:02:59 2025) ================ +================ Training Loss (Mon Feb 24 23:07:07 2025) ================ +(epoch: 1, iters: 100, time: 0.155, data: 0.229) G_GAN: 6.946 D_real_ViT: 0.807 D_fake_ViT: 0.309 G: 7.258 SB: 0.056 +(epoch: 1, iters: 200, time: 0.174, data: 0.004) G_GAN: 5.494 D_real_ViT: 0.538 D_fake_ViT: 0.287 G: 5.808 SB: 0.054 +(epoch: 1, iters: 300, time: 0.185, data: 0.003) G_GAN: 4.615 D_real_ViT: 0.424 D_fake_ViT: 0.302 G: 4.865 SB: 0.073 +(epoch: 1, iters: 400, time: 0.191, data: 0.003) G_GAN: 4.170 D_real_ViT: 0.357 D_fake_ViT: 0.315 G: 4.751 SB: 0.055 +(epoch: 1, iters: 500, time: 0.196, data: 0.004) G_GAN: 4.437 D_real_ViT: 0.323 D_fake_ViT: 0.265 G: 4.986 SB: 0.062 +(epoch: 1, iters: 600, time: 0.199, data: 0.003) G_GAN: 4.024 D_real_ViT: 0.289 D_fake_ViT: 0.252 G: 4.247 SB: 0.073 +(epoch: 1, iters: 700, time: 0.202, data: 0.004) G_GAN: 3.999 D_real_ViT: 0.273 D_fake_ViT: 0.232 G: 4.461 SB: 0.058 +(epoch: 1, iters: 800, time: 0.204, data: 0.003) G_GAN: 3.754 D_real_ViT: 0.233 D_fake_ViT: 0.241 G: 4.065 SB: 0.054 +(epoch: 1, iters: 900, time: 0.205, data: 0.003) G_GAN: 4.083 D_real_ViT: 0.249 D_fake_ViT: 0.207 G: 4.281 SB: 0.059 diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch001_fake_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch001_fake_B0.png new file mode 100644 index 0000000..09c5f95 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch001_fake_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A0.png new file mode 100644 index 0000000..c59b80f Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A_noisy.png b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A_noisy.png new file mode 100644 index 0000000..e6eb10c Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A_noisy.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_B0.png new file mode 100644 index 0000000..cd56419 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/index.html b/checkpoints/ROMA_UNSB_001/web/index.html new file mode 100644 index 0000000..b199ea4 --- /dev/null +++ b/checkpoints/ROMA_UNSB_001/web/index.html @@ -0,0 +1,45 @@ + + + + Experiment name = ROMA_UNSB_001 + + +

epoch [1]

+ + + + + + + +
+

+ + +
+

real_A0

+

+ 		+

+ + +
+

real_A_noisy

+

+ 		+

+ + +
+

fake_B0

+

+ 		+

+ + +
+

real_B0

+

+
+ + \ No newline at end of file diff --git a/models/__pycache__/roma_unsb_model.cpython-39.pyc b/models/__pycache__/roma_unsb_model.cpython-39.pyc index 15c9868..ae5f110 100644 Binary files a/models/__pycache__/roma_unsb_model.cpython-39.pyc and b/models/__pycache__/roma_unsb_model.cpython-39.pyc differ diff --git a/models/roma_unsb_model.py b/models/roma_unsb_model.py index 497ec9c..68ca586 100644 --- a/models/roma_unsb_model.py +++ b/models/roma_unsb_model.py @@ -239,9 +239,8 @@ class RomaUnsbModel(BaseModel): BaseModel.__init__(self, opt) # 指定需要打印的训练损失 - self.loss_names = ['G_GAN_1', 'D_real_1', 'D_fake_1', 'G_1', 'NCE_1', 'SB_1', - 'G_2'] - self.visual_names = ['real_A', 'real_A_noisy', 'fake_B', 'real_B'] + self.loss_names = ['G_GAN', 'D_real_ViT', 'D_fake_ViT', 'G', 'SB'] + self.visual_names = ['real_A0', 'real_A_noisy', 'fake_B0', 'real_B0'] self.atten_layers = [int(i) for i in self.opt.atten_layers.split(',')] if self.opt.phase == 'test':