Skip to content
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

Bnb/training with obs #252

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

Bnb/training with obs #252

wants to merge 32 commits into from

Conversation

bnb32
Copy link
Collaborator

@bnb32 bnb32 commented Dec 27, 2024

Added observation member to Sup3rDataset and minor modifications to samplers / batch queues to enable additional observation based loss term in training. Subclassed Sup3rGan to create Sup3rGanWithObs model. Refactored some model code to reduce duplication with new model.

@bnb32 bnb32 force-pushed the bnb/training_with_obs branch 4 times, most recently from bd493b1 to 7897998 Compare January 2, 2025 20:05
@bnb32 bnb32 force-pushed the bnb/training_with_obs branch 3 times, most recently from c14708f to 99cbac5 Compare January 8, 2025 21:01
bnb32 added 23 commits January 19, 2025 06:44
…ind model training added to examples README.
…`` class to enable passing ``xr.Dataset`` and like objects directly to ``BatchHandlers`` without needing to invoke ``.data``
…ueueing, since batches can be sampled directly if there are none in the queue).
…d dequeueing since this would cast to tensors.
… just sampling is, except for macos, but training is not.
@bnb32 bnb32 force-pushed the bnb/training_with_obs branch from 99cbac5 to 5afa9ed Compare January 19, 2025 13:51
bnb32 added 4 commits January 19, 2025 09:54
…ate models with increased receptive field and larger padding values.
…adding`` layers in the generator model. Generalized current min padding to use these values.
@bnb32 bnb32 marked this pull request as ready for review January 21, 2025 13:33
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