123456/roma_unsb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

http://47.108.14.56:3000/c/b78fb9d0-d385-4dce-81dd-5173e1cf037a

Browse Source
This commit is contained in:
bishe 2025-02-23 16:44:28 +08:00
parent c0364b0170
commit 9966211e8b
4 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
checkpoints/ROMA_UNSB_001/loss_log.txt
View File

@ -4,3 +4,4 @@
================ Training Loss (Sun Feb 23 16:02:40 2025) ================
================ Training Loss (Sun Feb 23 16:05:19 2025) ================
================ Training Loss (Sun Feb 23 16:06:44 2025) ================
================ Training Loss (Sun Feb 23 16:09:38 2025) ================

BIN
models/__pycache__/roma_unsb_model.cpython-39.pyc
View File

Binary file not shown.

2
models/networks.py
View File

@ -984,7 +984,7 @@ class ResnetGenerator(nn.Module):
else:
# print("%d: skipping %s %d" % (layer_id, layer.__class__.__name__, feat.size(1)))
pass
if layer_id == layers[-1] and encode_only:
if layer_id == len(list(self.model)) - 1 and encode_only:
# print('encoder only return features')
return feats # return intermediate features alone; stop in the last layers

1
models/roma_unsb_model.py
View File

@ -405,7 +405,6 @@ class RomaUnsbModel(BaseModel):
self.mutil_real_A0_tokens = self.netPreViT(real_A0, self.atten_layers, get_tokens=True)
self.mutil_real_A1_tokens = self.netPreViT(real_A1, self.atten_layers, get_tokens=True)
print(self.mutil_real_A0_tokens)
self.mutil_real_A0_tokens = torch.cat(self.mutil_real_A0_tokens, dim=1).to(self.device)
self.mutil_real_A1_tokens = torch.cat(self.mutil_real_A1_tokens, dim=1).to(self.device)