Open EXing opened 1 year ago
ht, wd = disps.shape[2:] should be ht, wd = disps.shape[-2:]
ht, wd = disps.shape[2:]
ht, wd = disps.shape[-2:]
ht, wd = disps.shape[2:]
should beht, wd = disps.shape[-2:]