# NN-architecture and logic parameters


NNConfig.dropout_rate = 0.1
NNConfig.hidden_act = "gelu"
NNConfig.layer_norm_eps = 1e-12
NNConfig.scale = True
NNConfig.use_scale_norm = False
NNConfig.use_adasoftmax = False
#
NNConfig.model_name = 'edgenext_base_usi'
NNConfig.path2ckpt = '/in/models/weights/'
#
NNConfig.device = 'cuda'