mirror of
https://github.com/helblazer811/ManimML.git
synced 2025-05-20 12:05:58 +08:00
Added sigmoid activation, and ability to do activation above linear layers.
This commit is contained in:
@ -68,6 +68,11 @@ class Convolutional2DLayer(VGroupNeuralNetworkLayer, ThreeDLayer):
|
||||
about_point=self.get_center(),
|
||||
axis=ThreeDLayer.rotation_axis,
|
||||
)
|
||||
|
||||
self.construct_activation_function()
|
||||
|
||||
def construct_activation_function(self):
|
||||
"""Construct the activation function"""
|
||||
# Add the activation function
|
||||
if not self.activation_function is None:
|
||||
# Check if it is a string
|
||||
|
Reference in New Issue
Block a user