Scene-Graph-Benchmark.pytorch icon indicating copy to clipboard operation
Scene-Graph-Benchmark.pytorch copied to clipboard

Question about pre-trained model

Open TungWg opened this issue 4 years ago • 0 comments

❓ Questions and Help

Hi Kaihua! Thank you for your excellent work. I checked the network, I found the parameter.requires_grad==True for roi_heads.relation.union_feature_extractor and roi_heads.relation.box_feature_extractor, Does this mean that these two modules need training? Do these two modules have pre-trained parameters? thanks!

TungWg avatar Oct 31 '21 03:10 TungWg