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