You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
1000 B
17 lines
1000 B
data: 'dataset/test' # path to dataset
|
|
output_dir: 'output/' # path to output files
|
|
model: 'tf_efficientnet_b5' # model architecture (default: dpn92)
|
|
workers: 4 # number of McD loading workers (default: 2
|
|
batch_size: 256 # mini-batch size (default: 256)
|
|
img_size: 224 # Input image dimension
|
|
mean: null # Override mean pixel value of dataset
|
|
std: null # Override std deviation of of dataset
|
|
interpolation: '' # Image resize interpolation type (overrides model)
|
|
num_classes: 1000 # Number classes in dataset
|
|
log_freq: 10 # batch logging frequency (default: 10)
|
|
checkpoint: '' # path to latest checkpoint (default: none)
|
|
pretrained: True # use pre-trained model
|
|
num_gpu: 1 # Number of GPUS to use
|
|
no_test_pool: False # disable test time pool
|
|
topk: 5 # Top-k to output to CSV
|