52 lines
965 B
YAML
52 lines
965 B
YAML
name: cross-domain-1
|
|
engine: crossdomain
|
|
result_dir: ./result
|
|
|
|
distributed:
|
|
model:
|
|
# broadcast_buffers: False
|
|
|
|
misc:
|
|
random_seed: 1004
|
|
|
|
checkpoints:
|
|
interval: 2000
|
|
|
|
log:
|
|
logger:
|
|
level: 20 # DEBUG(10) INFO(20)
|
|
|
|
model:
|
|
_type: resnet10
|
|
|
|
baseline:
|
|
plusplus: False
|
|
optimizers:
|
|
_type: Adam
|
|
data:
|
|
dataloader:
|
|
batch_size: 1200
|
|
shuffle: True
|
|
num_workers: 16
|
|
pin_memory: True
|
|
drop_last: True
|
|
dataset:
|
|
train:
|
|
path: /data/few-shot/mini_imagenet_full_size/train
|
|
lmdb_path: /data/few-shot/lmdb/mini-ImageNet/train.lmdb
|
|
pipeline:
|
|
- Load
|
|
- RandomResizedCrop:
|
|
size: [224, 224]
|
|
- ColorJitter:
|
|
brightness: 0.4
|
|
contrast: 0.4
|
|
saturation: 0.4
|
|
- RandomHorizontalFlip
|
|
- ToTensor
|
|
- Normalize:
|
|
mean: [0.485, 0.456, 0.406]
|
|
std: [0.229, 0.224, 0.225]
|
|
|
|
|