KeyError: ‘fasterrcnn_resnet50’

[abubakarsaleem66]

I am trying to run this GitHub repository faster rcnn-pytorch-custom-dataset but I got this error.

Building model from scratch...
Traceback (most recent call last):
  File "train.py", line 491, in <module>
    main(args)
  File "train.py", line 248, in main
    build_model = create_model[args['model']]
KeyError: 'fasterrcnn_resnet50'
wandb: Waiting for W&B process to finish... (failed 1). Press Control-C to abort syncing.
wandb: Synced smoke_training: https://wandb.ai/samahwa/fastercnn-pytorch-training-pipeline/runs/ejy5jyw8
wandb: Synced 5 W&B file(s), 0 media file(s), 0 artifact file(s) and 0 other file(s)
wandb: Find logs at: ./wandb/run-20221128_113545-ejy5jyw8/logs

[bill-morrisson]

Hello Abubakar,

Thank you for your patience and sorry this is happening to you. Could you please let me know;

What is your wandb cli version?
Are there any other errors shown except these?
Could you please share with me the debug.log and debug-internal.log files located in the working directory of the project under wandb//logs/?
Are you running your script in jupyter notebook or terminal or through colab, etc.? Can I have an example training script of what you did exactly?

Thank you

[bill-morrisson]

Hello Abubakar,

Could you please get back to me with an answer to my questions? That would enable me to have more information on what might be going wrong.