# training
data_dir = '/playpen/connylu/gaze_object_detection/dataset'
model_name = 'inception'
# num_classes = 25  # data_random
num_classes = 39  # realisticrendering_extraprops
batch_size = 8
num_epochs = 200
feature_extract = True

lr = 0.001
momentum = 0.9
