opensource/ManimML
1
0
Fork 0
mirror of https://github.com/helblazer811/ManimML.git synced 2025-08-06 00:48:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added ability to pass layer_args dictionary to each forward pass, which allows

Browse Source 
for arguments to be passed through to each neural network layer when running a neural
network forward pass.
This commit is contained in:
Alec Helbling
2022-04-25 16:28:11 -04:00
parent 7d04bf55ec
commit 63427be139
18 changed files with 145 additions and 282 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
manim_ml/neural_network/layers/feed_forward.py
View File

@ -44,7 +44,7 @@ class FeedForwardLayer(VGroupNeuralNetworkLayer):
# Add the objects to the class
self.add(self.surrounding_rectangle, self.node_group)
def make_forward_pass_animation(self, **kwargs):
def make_forward_pass_animation(self, layer_args={}, **kwargs):
# make highlight animation
succession = Succession(
ApplyMethod(self.node_group.set_color, self.animation_dot_color, run_time=0.25),