TRI-ML / vidar

Other
554 stars 65 forks source link

support for batch size > 1 #28

Open jonasitzmann opened 1 year ago

jonasitzmann commented 1 year 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