2021-12-16 02:32:08 +00:00
|
|
|
import os
|
2021-12-29 15:32:10 +00:00
|
|
|
import model
|
2021-12-16 02:32:08 +00:00
|
|
|
from pathlib import Path
|
|
|
|
|
|
|
|
|
|
BATCH_SIZE = 128
|
|
|
|
|
IMG_SIZE = 224
|
|
|
|
|
DIM = 768
|
|
|
|
|
NUM_CLASSES = 10
|
|
|
|
|
NUM_ATTN_HEADS = 12
|
2022-04-03 12:46:45 +00:00
|
|
|
NUM_MICRO_BATCHES = 2
|
2021-12-16 02:32:08 +00:00
|
|
|
|
|
|
|
|
# resnet 18
|
|
|
|
|
model = dict(type='VanillaResNet',
|
|
|
|
|
block_type='ResNetBasicBlock',
|
|
|
|
|
layers=[2, 2, 2, 2],
|
|
|
|
|
num_cls=10)
|
|
|
|
|
|
|
|
|
|
parallel = dict(
|
|
|
|
|
pipeline=dict(size=4),
|
|
|
|
|
tensor=dict(size=1, mode=None)
|
|
|
|
|
)
|
