Skip to content

GitLab

"driver/conv_driver.cpp" did not exist on "c15ff3c825dd25ddc2f6ee572f867c925a4528bd"
Commit 5cd237dc authored by renzhc's avatar renzhc
Browse files

fixed efficientnetb2

parent 12c02d09
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 25 deletions
configs/_base_/models/tiny_efficientnet_b2.py
View file @ 5cd237dc
_base_ = [
'configs/_base_/models/tiny_efficientnet_b2.py',
'configs/_base_/datasets/tiny_imagenet_bs32.py',
'configs/_base_/schedules/imagenet_bs256.py',
'configs/_base_/default_runtime.py',
]
# dataset settings
train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='EfficientNetRandomCrop', scale=260),
dict(type='RandomFlip', prob=0.5, direction='horizontal'),
dict(type='PackInputs'),
]
test_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='EfficientNetCenterCrop', crop_size=260),
dict(type='PackInputs'),
]
train_dataloader = dict(dataset=dict(pipeline=train_pipeline))
val_dataloader = dict(dataset=dict(pipeline=test_pipeline))
test_dataloader = dict(dataset=dict(pipeline=test_pipeline))
root@K100_AI02:/renzhc/workdir/mmpretrain# cat configs/_base_/models/tiny_efficientnet_b2.py
# model settings
model = dict(
type='ImageClassifier',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment