search

yrcong / RelTR

RelTR: Relation Transformer for Scene Graph Generation: https://arxiv.org/abs/2201.11460v2
229 stars 45 forks source link

you commented out training code, training does not work at the moment #25

Closed altansnl closed 7 months ago

altansnl commented 1 year ago

https://github.com/yrcong/RelTR/commit/4d676ee5e3f8d01cd8a12648fdfb395dd6d17cdd

yrcong commented 1 year ago

Could you please give the error output?

altansnl commented 1 year ago

i do not have it at the moment, but you commented out the line we calls train_one_epoch in main.py

Vincent-luo commented 1 year ago

Yeah in the latest commit, in main.py lines including train_one_epoch are commented. So if I want to do training, all I have to do is to uncomment these lines?

altansnl commented 12 months ago

yeah you can uncomment and it works. i have been training a lot of RelTR.