Is there any scripts relatable to training are available?

isl-org / DPT

Dense Prediction Transformers

MIT License

1.96k stars 254 forks source link

Is there any scripts relatable to training are available? #54

Open InfiniteLife opened 2 years ago

InfiniteLife commented 2 years ago

I've seen few issues where people were trying to train the network, but cant find any training code. Is there anything available? I'm interested in loss function, pipeline for data and depth preparation for loss function, it seems rather complicated from reading the paper.

younesbelkada commented 2 years ago

Hi! together with my colleague we made an attempt to reproduce the training script here: https://github.com/antocad/FocusOnDepth check it out if you are interested!