mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #147 from littletomatodonkey/fix_effnet

Browse Source 
fix EfficientNet
This commit is contained in:
littletomatodonkey 2020-06-08 16:34:11 +08:00 committed by GitHub
commit c2034c50c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
configs/EfficientNet/EfficientNetB0.yaml
View File

@ -1,8 +1,12 @@
mode: 'train' mode: 'train'
ARCHITECTURE: ARCHITECTURE:
name: "EfficientNetB0" name: "EfficientNetB0"
drop_connect_rate: 0.1 params:
padding_type : "SAME" is_test: False
padding_type : "SAME"
override_params:
drop_connect_rate: 0.1
pretrained_model: "" pretrained_model: ""
model_save_dir: "./output/" model_save_dir: "./output/"
classes_num: 1000 classes_num: 1000

4
tools/program.py
View File

@ -103,7 +103,8 @@ def create_model(architecture, image, classes_num, is_train):
""" """
name = architecture["name"] name = architecture["name"]
params = architecture.get("params", {}) params = architecture.get("params", {})
params['is_test'] = not is_train if "is_test" in params:
params['is_test'] = not is_train
model = architectures.__dict__[name](**params) model = architectures.__dict__[name](**params)
out = model.net(input=image, class_dim=classes_num) out = model.net(input=image, class_dim=classes_num)
return out return out
@ -418,7 +419,6 @@ def run(dataloader,
Returns: Returns:
""" """
print(fetchs)
fetch_list = [f[0] for f in fetchs.values()] fetch_list = [f[0] for f in fetchs.values()]
metric_list = [f[1] for f in fetchs.values()] metric_list = [f[1] for f in fetchs.values()]
for m in metric_list: for m in metric_list: