add wandb, make compatible with ctd data

2022-05-30 16:13:06 +02:00
parent b333d5515f
commit 70e4bf6fe1
4 changed files with 471 additions and 44 deletions
@@ -155,6 +155,8 @@ class CREStereo(nn.Module):
        flow = None
        flow_up = None
        if flow_init is not None:
+            if isinstance(flow_init, list):
+                flow_init = flow_init[0]
            scale = fmap1.shape[2] / flow_init.shape[2]
            flow = -scale * F.interpolate(
                flow_init,