Giters

weders / RoutedFusion

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Wrong channel

eriksandstroem opened this issue · comments

commented

RoutedFusion/train_routing.py

Line 86 in 110900b

est = output[:, 0, :, :].unsqueeze_(1)

commented

Actually, this is wrong. Line 87 should be changed instead.