hi-ml
hi-ml copied to clipboard
Add documentation to show how multi-node training works
- Pass num_nodes to submit_to_azure_if_needed
- Pass it also to the PL Trainer object
- After the call to submit_to_azure_if_needed, call set_environment_variables_for_multi_node