mirrors/mmpretrain
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmpretrain.git synced 2025-06-03 14:59:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #53 from daavoo/fix-configs

Browse Source 
fix outdated configs
This commit is contained in:
Y. Xiong 2020-10-01 23:05:56 +08:00 committed by GitHub
commit 52d8ea4621
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
configs/_base_/datasets/cifar10_bs16.py
View File

@ -15,8 +15,7 @@ train_pipeline = [
test_pipeline = [
dict(type='Normalize', **img_norm_cfg),
dict(type='ImageToTensor', keys=['img']),
dict(type='ToTensor', keys=['gt_label']),
dict(type='Collect', keys=['img', 'gt_label'])
dict(type='Collect', keys=['img'])
]
data = dict(
samples_per_gpu=16,

3
docs/tutorials/data_pipeline.md
View File

@ -31,8 +31,7 @@ test_pipeline = [
dict(type='CenterCrop', crop_size=224),
dict(type='Normalize', **img_norm_cfg),
dict(type='ImageToTensor', keys=['img']),
dict(type='ToTensor', keys=['gt_label']),
dict(type='Collect', keys=['img', 'gt_label'])
dict(type='Collect', keys=['img'])
]
```

3
docs/tutorials/finetune.md
View File

@ -59,8 +59,7 @@ train_pipeline = [
dict(type='Resize', size=224)
dict(type='Normalize', **img_norm_cfg),
dict(type='ImageToTensor', keys=['img']),
dict(type='ToTensor', keys=['gt_label']),
dict(type='Collect', keys=['img', 'gt_label'])
dict(type='Collect', keys=['img'])
]
```

2
setup.cfg
View File

@ -17,6 +17,6 @@ line_length = 79
multi_line_output = 0
known_standard_library = pkg_resources,setuptools
known_first_party = mmcls
known_third_party = PIL,cv2,mmcv,numpy,pytest,torch,torchvision
known_third_party = PIL,cv2,mmcv,numpy,onnxruntime,pytest,torch,torchvision
no_lines_before = STDLIB,LOCALFOLDER
default_section = THIRDPARTY