14 lines
329 B
Plaintext
14 lines
329 B
Plaintext
# 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' |