Regarding Pretrained Model Weights

yrcong / RelTR

RelTR: Relation Transformer for Scene Graph Generation: https://arxiv.org/abs/2201.11460v2

230 stars 45 forks source link

Regarding Pretrained Model Weights #2

Closed hyzcn closed 2 years ago

hyzcn commented 2 years ago

Dear Author, thanks for your fabulous code! I have a question regarding model training, wondering if you initialized the model with pretrained detr model or you just train the whole network from scratch?

yrcong commented 2 years ago

I trained the whole network from scratch:)

hyzcn commented 2 years ago

Thanks for your reply @yrcong . Wondering how long will it take a lot time to converge. I tried to train on my data, but seems very slow. Wondering how many epoch typically need

yrcong commented 2 years ago

I remembered I have written in the paper. I trained 150 epochs with 8 RTX2080 (2 images per GPU).