Merge pull request #1527 from klae01/patch-1

Update train.py
2 years ago · f856f8a2e0
parent 4201c718e9 ddd6361904
commit f856f8a2e0
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -407,7 +407,7 @@ def main():

    in_chans = 3
    if args.in_chans is not None:
-        in_chans = args.in_chanes
+        in_chans = args.in_chans
    elif args.input_size is not None:
        in_chans = args.input_size[0]