Skip to content

Avalon-467/ai_model

Repository files navigation

ai_model

a repo collects different ai models
人工智能模型集
index:

base_model

customized_MLP: a mlp model which user can define the number of nodes in each layer of mlp freely and easily just by a list is composed of int numbers.
通过整数列表便捷定义全连接层的各层节点数目。

customized_conv: a multi_layers convolution model, which user can define it by a list of numbers
通过整数列表便捷定义卷积层的各层节点数目

reset_layer: partition and shuffer the feature map of conv
分块打散特征图,以令卷积获得全局建模

train_test

trainer: train the specific model in specific trian_dataset
通用训练器:指定模型和训练集

classifier_tester: a tester for classifier task
分类任务测试器

About

a repo collects different ai model

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages