rs.   is_main_process (bool, optional, defaults to True) —
Whether the process calling this is the main process or not. Useful during distributed training and you
need to call this function on all processes. In this case, set is_main_process=True only on the main
process to avoid race conditions.   save_function (Callable) —
The function to use to save the state dictionary. Useful during distributed training when you need to
replace torch.save 