参考鱼书写的一个完全由numpy实现的深度学习框架(加速用的cupy),目前实现了比如:activation function: sigmoid / relu / softmax / log_softmax / leaky_relu / mish / silu ;loss function: mean_squared_error / softmax_cross_entropy / sigmoid_cross_entropy / binary_cross_entropy ; dropout / batch_norm; 2d卷积; BatchNorm / MaxPool2d / LayerNormalization / GroupNorm ; 注意力机制(没测试);以及一个魔改的yolo模型
-
Notifications
You must be signed in to change notification settings - Fork 0
参考鱼书写的一个完全由numpy实现的深度学习框架(加速用的cupy),目前实现了比如:activation function: sigmoid / relu / softmax / log_softmax / leaky_relu / mish / silu ;loss function: mean_squared_error / softmax_cross_entropy / sigmoid_cross_entropy / binary_cross_entropy ; dropout / batch_norm; 2d卷积; BatchNorm / MaxPool2d / LayerNormalization / GroupNorm ; 注意力机制(没测试);以及一个魔改的yolo模型
License
zgh2022/project_samsara
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
参考鱼书写的一个完全由numpy实现的深度学习框架(加速用的cupy),目前实现了比如:activation function: sigmoid / relu / softmax / log_softmax / leaky_relu / mish / silu ;loss function: mean_squared_error / softmax_cross_entropy / sigmoid_cross_entropy / binary_cross_entropy ; dropout / batch_norm; 2d卷积; BatchNorm / MaxPool2d / LayerNormalization / GroupNorm ; 注意力机制(没测试);以及一个魔改的yolo模型
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published