DimTrigkakis:

pytorch


I think you mean the flownet model. It has quite some pytorch implementation

  
      
      GitHub
  
  
    

ClementPinard/FlowNetPytorch 272

Pytorch implementation of FlowNet by Dosovitskiy et al. - ClementPinard/FlowNetPytorch