Skip to content

Add support for loading .npy dataset#227

Open
shrutiishar wants to merge 2 commits intoML4SCI:mainfrom
shrutiishar:main
Open

Add support for loading .npy dataset#227
shrutiishar wants to merge 2 commits intoML4SCI:mainfrom
shrutiishar:main

Conversation

@shrutiishar
Copy link
Copy Markdown

Added a custom PyTorch Dataset class to load .npy-based lensing data.

This enables training models directly on DeepLense datasets without converting them into image format.

Tested with a ResNet18 baseline model.

Would appreciate feedback and suggestions.

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