fix(train.py): typeError: build_model() got an unexpected keyword arg…

…ument 'backbone'
zjykzj · Sep 22, 2022 · 663fa1f · 663fa1f
1 parent 4a05509
commit 663fa1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/multigrain/train.py b/multigrain/train.py
@@ -256,7 +256,7 @@ def main(args):
 
     print("Creating model")
     # model = torchvision.models.get_model(args.model, weights=args.weights, num_classes=num_classes)
-    model = build_model(backbone=args.model)
+    model = build_model(args)
     model.to(device)
 
     if args.distributed and args.sync_bn: