""" configurations for this project

author baiyu
"""
import os
from datetime import datetime

#CIFAR100 dataset path (python version)
#CIFAR100_PATH = '/nfs/private/cifar100/cifar-100-python'

#mean and std of cifar100 dataset
# CIFAR100_TRAIN_MEAN = (0.5070751592371323, 0.48654887331495095, 0.4409178433670343)
# CIFAR100_TRAIN_STD = (0.2673342858792401, 0.2564384629170883, 0.27615047132568404)

# CIFAR100_TEST_MEAN = (0.5088964127604166, 0.48739301317401956, 0.44194221124387256)
# CIFAR100_TEST_STD = (0.2682515741720801, 0.2573637364478126, 0.2770957707973042)

#mean and std of GTSRB in 32by32 size dataset
# CIFAR100_TRAIN_MEAN = (0.338427106020795, 0.3113211112064918, 0.3199274353230347)
# CIFAR100_TRAIN_STD = (0.27344659720932935, 0.2643566180588379, 0.2681631351408068)
CIFAR100_TRAIN_MEAN = (0.5, 0.5, 0.5)
CIFAR100_TRAIN_STD = (0.5, 0.5, 0.5)

# CIFAR100_TEST_MEAN = (0.3357295806967825, 0.30876526367900975, 0.3192921114021899)
# CIFAR100_TEST_STD = (0.27072321025572466, 0.26269992754359084, 0.26944605745028744)



#directory to save weights file
CHECKPOINT_PATH = 'checkpoint'

#total training epoches
EPOCH = 200
MILESTONES = [60, 120, 160]

#initial learning rate
#INIT_LR = 0.1

DATE_FORMAT = '%A_%d_%B_%Y_%Hh_%Mm_%Ss'
#time of we run the script
TIME_NOW = datetime.now().strftime(DATE_FORMAT)

#tensorboard log dir
LOG_DIR = 'runs'

#save weights file per SAVE_EPOCH epoch
SAVE_EPOCH = 10








