When I run your script, at the line of
demo = VisualizationDemo(cfg)
It will try to print out the entire model like:
....
| roi_heads.text_decoder.textual.transformer.encoder.layer.5.intermediate.dense.* | roi_heads.text_decoder.textual.transformer.encoder.layer.5.intermediate.dense.{bias,weight}
....
When I run your script, at the line of
demo = VisualizationDemo(cfg)
It will try to print out the entire model like: .... | roi_heads.text_decoder.textual.transformer.encoder.layer.5.intermediate.dense.* | roi_heads.text_decoder.textual.transformer.encoder.layer.5.intermediate.dense.{bias,weight} ....