Learning only on first sample of the batch

ClementPinard / SfmLearner-Pytorch

Pytorch version of SfmLearner from Tinghui Zhou et al.

MIT License

1.01k stars 226 forks source link

Learning only on first sample of the batch #77

Closed kristijanbartol closed 4 years ago

kristijanbartol commented 4 years ago

Hi Clement,

thank you for this nice implementation, it is very useful. I am using it to propose an improvement over pixel reprojection methods and as a base for other ideas on SfM problem.

I came across these lines and I wonder why did you choose to use only a first sample of the batch? Or am I missing something?

Thank you, Kristijan

kristijanbartol commented 4 years ago

I'm sorry, I realized you are using this warped image only to log it to TensorBoard and learn on the whole batch. My problem is regarding SSIM loss implementation and particular dimensions' problem, but I think I will be able to cope with it. Thank you!