# file: $(NEDC_NFC)/util/python/nedc_dpath_tools/nedc_dpath_train_param.py
#
# A parameter file for training in DPATH systems, especially desigened for
# training the ResNet18.
#
# revision history:
#
# 20210325 (VK): make it campatible to NEDC parameter files format
# 20210221 (VK): dev and eval sets can be selective
# 20210128 (VK): second version
# 20200904 (VK): initial version
#
# set version information
#
version = param_v1.0.0

#------------------------------------------------------------------------------

TRAIN {

    # batch size
    #
    bat_size = 8
    #bat_size = 2

    # number of epochs
    #
    #num_epochs = 3
    num_epochs = 25

    # number of samples in each class
    #
    # nsamples_per_class = 5000
    # nsamples_per_class = {
    #     'train': 100000,
    #     'dev': 100000
    #     }
    nsamples_train = 36
    nsamples_dev = 18

    # number of workers to load data into memory
    #
    num_workers = 8

    # GPU or CPU device
    #
    device = 'cuda:0'
    #device = 'cuda:2'
    # device = 'cpu'
}

#------------------------------------------------------------------------------
#
# end of file
#
#------------------------------------------------------------------------------
