mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add in_chans to data config helper. Fix #2021

Browse Source
This commit is contained in:
Ross Wightman 2023-11-23 12:43:03 -08:00
parent 63ee54853c
commit 40d55ab4bc
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/data/config.py
View File

@ -21,7 +21,9 @@ def resolve_data_config(
# Resolve input/image size
in_chans = 3
if args.get('chans', None) is not None:
if args.get('in_chans', None) is not None:
in_chans = args['in_chans']
elif args.get('chans', None) is not None:
in_chans = args['chans']
input_size = (in_chans, 224, 224)