search

yrcong / RelTR

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

Will RelTR support Resnet101 as backbone and provided a trained checkpoint with Resnet101 as backbone? #8

Closed linhuixiao closed 1 year ago

linhuixiao commented 1 year ago

Will RelTR support Resnet101 as backbone and provided a trained checkpoint with Resnet101 as backbone?

yrcong commented 1 year ago

Regardless of GPU memory constraint, ResNet101 can be used definitely. You can directly use the pretrained ResNet101 provided by Pytorch, as we used Res50.