TRI-ML / vidar

Other
573 stars 67 forks source link

support for batch size > 1 #28

Open jonasitzmann opened 2 years ago

jonasitzmann commented 2 years ago

Hi, When training the depthformer model a batch size > 1 yields to several errors. I am getting the impression that the code is not meant to handle any other batch sizes (except for data parallel training with mutiple gpus) Or am I missing something? Thanks for your work and best regards, Jonas