Skip to content

Conversation

@jvdp1
Copy link
Collaborator

@jvdp1 jvdp1 commented Apr 21, 2024

No description provided.

@jvdp1 jvdp1 marked this pull request as ready for review April 21, 2024 11:56
Comment on lines 600 to 604
call random_number(pos)
batch_start = int(pos * (dataset_size - batch_size + 1)) + 1
batch_end = batch_start + batch_size - 1

! FIXME shuffle in a way that doesn't require co_broadcast
call co_broadcast(batch_start, 1)
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This could be called outside the loop by creating an array batch_start(:) and by co_broadcasting it only once

@milancurcic milancurcic self-requested a review May 9, 2024 18:26
Copy link
Member

@milancurcic milancurcic left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you!

@milancurcic milancurcic merged commit a843c83 into modern-fortran:main Jun 14, 2024
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.

2 participants