KeyError: ‘fasterrcnn_resnet50’

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

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

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.

This topic was automatically closed 60 days after the last reply. New replies are no longer allowed.