diff --git a/manim_ml/neural_network/neural_network.py b/manim_ml/neural_network/neural_network.py index 1232d03..8fe0cd4 100644 --- a/manim_ml/neural_network/neural_network.py +++ b/manim_ml/neural_network/neural_network.py @@ -331,7 +331,7 @@ def make_forward_pass_animation( # Add the animation to per layer animation per_layer_animation_map[layer] = layer_forward_pass # Make the animation group - animation_group = Succession(*all_animations, lag_ratio=1.0) + animation_group = Succession(*all_animations, lag_ratio=1.0, run_time=run_time) if per_layer_animations: return per_layer_animation_map else: