Skip to content

Reduce the batch size in the TensorFlow RN50 L3 test #5565

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

Merged
merged 1 commit into from
Jul 17, 2024

Conversation

JanuszL
Copy link
Contributor

@JanuszL JanuszL commented Jul 15, 2024

  • due to an increase in the GPU memory consumption in the latest
    TensorFlow container the batch size needs to be reduced
    to fit the available memory in some test runners

Category:

Other (e.g. Documentation, Tests, Configuration)

Description:

  • due to an increase in the GPU memory consumption in the latest
    TensorFlow container the batch size needs to be reduced
    to fit the available memory in some test runners

Additional information:

Affected modules and functionalities:

  • TL3_RN50_convergence

Key points relevant for the review:

  • NA

Tests:

  • Existing tests apply
  • New tests added
    • Python tests
      • TL3_RN50_convergence/test_tensorflow.sh
    • GTests
    • Benchmark
    • Other
  • N/A

Checklist

Documentation

  • Existing documentation applies
  • Documentation updated
    • Docstring
    • Doxygen
    • RST
    • Jupyter
    • Other
  • N/A

DALI team only

Requirements

  • Implements new requirements
  • Affects existing requirements
  • N/A

REQ IDs: N/A

JIRA TASK: N/A

- due to an increase in the GPU memory consumption in the latest
  TensorFlow container the batch size needs to be reduced
  to fit the available memory in some test runners

Signed-off-by: Janusz Lisiecki <[email protected]>
@dali-automaton
Copy link
Collaborator

CI MESSAGE: [16618389]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [16618389]: BUILD PASSED

@JanuszL JanuszL merged commit b0b0241 into NVIDIA:main Jul 17, 2024
6 checks passed
@JanuszL JanuszL deleted the tf_rn50_batch branch July 17, 2024 07: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.

4 participants