search

sniklaus / pytorch-pwc

a reimplementation of PWC-Net in PyTorch that matches the official Caffe version
GNU General Public License v3.0
616 stars 123 forks source link

about training #17

Closed ufukefe closed 5 years ago

ufukefe commented 5 years ago

Hi, is it possible to train this network with your code on a dataset like Sintel?

sniklaus commented 5 years ago

The provided implementation is fully differentiable. As such, you just need to provide a data loader and implement a loss function. Feel free to fork this repository and see whether you can make it work. It would be nice to have a fork with a working training section that I could link to.