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
Hi @samahwa thank you for writing in. The KeyError seems to stem from the model’s name you’re passing in. In the GitHub repo you stored, there is a models directory where you could find the valid names. Could you please try if fasterrcnn_resnet50_fpn or fasterrcnn_resnet50_fpn_v2 would work for you?
Hi @samahwa since we haven’t heard back from you, I will close this ticket for now. If you still experience any issue running your code, please let us know and we will keep investigating.