Skip to content

Commit c397600

Browse files
baorepoPillar1989
authored andcommitted
configs: Separate the pretrain and cls dir
1 parent 6baa6a8 commit c397600

File tree

6 files changed

+20
-155
lines changed

6 files changed

+20
-155
lines changed

configs/mobilenetv4_imagenet.py

Lines changed: 0 additions & 11 deletions
This file was deleted.
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
from mmengine.config import read_base
2+
3+
with read_base():
4+
from ..datasets.lancedb_bs32 import *
5+
from .._base_.default_runtime import *
6+
from ..cls.timm_classify import *
7+
from .._base_.schedules.AdamW_linear_coslr_bs2048 import *
8+
9+
train_dataloader.batch_size=64
10+
auto_scale_lr = dict(base_batch_size=64)
11+
model.model_name="mobilenetv4_hybrid_medium"
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from mmengine.config import read_base
2+
3+
with read_base():
4+
from ..datasets.imagenet_bs32 import *
5+
from .._base_.default_runtime import *
6+
from ..cls.timm_classify import *
7+
from .._base_.schedules.sgd_linear_coslr_bs2048 import *
8+
9+
model.model_name = "resnet50"

configs/pretrain/rtmdet_nano_8xb256_600e_coco_1k.py

Lines changed: 0 additions & 135 deletions
This file was deleted.

configs/resnet50_imagenet.py

Lines changed: 0 additions & 9 deletions
This file was deleted.

0 commit comments

Comments
 (0)