Skip to content

ResNet MLX #67

Open
gabrielfnayres wants to merge 3 commits intoapple:mainfrom
gabrielfnayres:resnet_mlx_main
Open

ResNet MLX #67
gabrielfnayres wants to merge 3 commits intoapple:mainfrom
gabrielfnayres:resnet_mlx_main

Conversation

@gabrielfnayres
Copy link
Contributor

Key Architectural Changes

  • Unsqueeze Process
    Adjusted tensor dimension expansion logic to ensure proper shape compatibility between temporal embeddings and spatial feature maps.

  • Temporal Embedding Broadcast
    Modified broadcasting logic to align temporal embeddings with spatial dimensions without introducing shape mismatches.


Testing & Validation

  • Test Strategy
    Implemented straightforward test cases to verify:

    • Layer behavior consistency
    • Numerical stability across resolutions
  • Resolved Issues

    • Fixed GroupNorm portability issues related to:
      • Parameter initialization
      • Axis/dimension handling during inference
    • Validated edge cases for temporal embedding integration

Screenshot 2025-02-26 at 17 37 18

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant