Skip to content

ResNet MLX #67

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

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