Первый коммит: код классификатора наручных знаков (Naruto)
This commit is contained in:
14
in/config_files/nn.gin
Normal file
14
in/config_files/nn.gin
Normal file
@@ -0,0 +1,14 @@
|
||||
# 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'
|
||||
Reference in New Issue
Block a user