How do we train HourglassDecoder?

[yutanpo1227]

I want to train the ConvNext model, but the training script that is given only works with RAFTDecoder.
How to train the ConvNext model?

[zero0kiriyu]

I have the same question. Wonder how to predict normal map in HourglassDecoder and which loss function is used for its supervision

[YvanYin]

In our V1 (hourglass decoder), we did not enclose normal prediction tasks. If you need that, you can directly add a head to predict the normal, and follow ViT-models losses for training.