Update total/Model.py
Browse files- total/Model.py +1 -0
total/Model.py
CHANGED
|
@@ -69,6 +69,7 @@ class Combine(Reduction):
|
|
| 69 |
pass
|
| 70 |
|
| 71 |
def __call__(self, tensors: list[torch.Tensor]) -> torch.Tensor:
|
|
|
|
| 72 |
fg_all = torch.cat([p[:, 1:, ...] for p in tensors], dim=1)
|
| 73 |
sum_fg = fg_all.sum(dim=1, keepdim=True)
|
| 74 |
bg = (1.0 - sum_fg).clamp(min=1e-6)
|
|
|
|
| 69 |
pass
|
| 70 |
|
| 71 |
def __call__(self, tensors: list[torch.Tensor]) -> torch.Tensor:
|
| 72 |
+
print(tensors[0].shape, tensors[0].dtype)
|
| 73 |
fg_all = torch.cat([p[:, 1:, ...] for p in tensors], dim=1)
|
| 74 |
sum_fg = fg_all.sum(dim=1, keepdim=True)
|
| 75 |
bg = (1.0 - sum_fg).clamp(min=1e-6)
|