Skip to content

Add option to concat coords.#257

Open
EmmaRenauld wants to merge 2 commits intoscil-vital:masterfrom
EmmaRenauld:add_coords_to_input
Open

Add option to concat coords.#257
EmmaRenauld wants to merge 2 commits intoscil-vital:masterfrom
EmmaRenauld:add_coords_to_input

Conversation

@EmmaRenauld
Copy link
Copy Markdown
Collaborator

  • In batchLoaderOneInput: option to concat raw coordinates or relative coordinates (raw / volume_dim) to the input data.

  • Added the option directly in the associated model: MainModelOneInput.

  • Learn2track and Transformers access it. Changed the management of the number of input features in both. Fixed input size with CNN embedding in Transformers.

  • Fixed some docstrings along the way

@EmmaRenauld EmmaRenauld changed the title Add option to concat coords. Needed to modify Transformer's d_model m… Add option to concat coords. May 6, 2025
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