Skip to content

check dtype in test data#1248

Merged
mrariden merged 1 commit intomainfrom
bfloat16_train
Jun 23, 2025
Merged

check dtype in test data#1248
mrariden merged 1 commit intomainfrom
bfloat16_train

Conversation

@mrariden
Copy link
Copy Markdown
Collaborator

Missed converting images and labels to bfloat16 in the validation/test data step in the train.train_seg() function.

Now test data dtype is checked for compatibility with network and conversion is applied if necessary.

@mrariden mrariden merged commit 8ef8804 into main Jun 23, 2025
7 checks passed
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