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