463465810cz commited on
Commit ·
c7bec9e
1
Parent(s): 4ab8bd0
ICCV 2023
Browse filesFormer-commit-id: 18975b7a3c308e9f89daa3d6ebc417297398d164
basicsr/archs/dat_arch.py
CHANGED
|
@@ -839,5 +839,9 @@ if __name__ == '__main__':
|
|
| 839 |
split_size=[8,16],
|
| 840 |
).cuda().eval()
|
| 841 |
|
|
|
|
|
|
|
| 842 |
x = torch.randn((1, 3, height, width)).cuda()
|
| 843 |
-
x = model(x)
|
|
|
|
|
|
|
|
|
| 839 |
split_size=[8,16],
|
| 840 |
).cuda().eval()
|
| 841 |
|
| 842 |
+
print(height, width)
|
| 843 |
+
|
| 844 |
x = torch.randn((1, 3, height, width)).cuda()
|
| 845 |
+
x = model(x)
|
| 846 |
+
|
| 847 |
+
print(x.shape)
|