Add JAX-defined augmentation examples #5426
Conversation
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
|
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
|
!build |
|
CI MESSAGE: [14196087]: BUILD STARTED |
|
CI MESSAGE: [14196087]: BUILD PASSED |
|
View / edit / reply to this conversation on ReviewNB awolant commented on 2024-04-15T10:50:50Z Line #3. os.environ["CUDA_VISIBLE_DEVICES"] = "0" # the other process passes 1 Maybe you could add some explanation that in a real world example this would have been set up by a launching script or something like that. |
|
I think you should consider adding those files under:
|
|
View / edit / reply to this conversation on ReviewNB klecki commented on 2024-04-15T15:00:10Z that access samples -> accesses? |
|
View / edit / reply to this conversation on ReviewNB klecki commented on 2024-04-15T15:00:10Z , the other with the -> will process or |
|
I would also consider adding a runnable script containing the other process to the multipgpu example, starting it alongside the jupyter and testing that it actually works. |
…c example in ci Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
|
!build |
|
CI MESSAGE: [14254191]: BUILD STARTED |
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
|
!build |
|
CI MESSAGE: [14257313]: BUILD STARTED |
|
CI MESSAGE: [14257313]: BUILD FAILED |
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
|
!build |
|
CI MESSAGE: [14273394]: BUILD STARTED |
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
|
!build |
|
CI MESSAGE: [14274963]: BUILD STARTED |
Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
|
!build |
|
CI MESSAGE: [14277138]: BUILD STARTED |
|
CI MESSAGE: [14277138]: BUILD PASSED |
* Add JAX operator examples: introducion and multi-gpu * Fix typo in error message --------- Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
Category:
Other (e.g. Documentation, Tests, Configuration)
Description:
Adds a basic jax_function usage example and a more involved example presenting multi process shard-aware processing.
Additional information:
Affected modules and functionalities:
Key points relevant for the review:
Tests:
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: DALI-3925