Small edits (#819)

This commit is contained in:
pollfly
2024-04-03 12:06:28 +03:00
committed by GitHub
parent 3ad296ab0b
commit 6e20f6ab0e
5 changed files with 15 additions and 11 deletions

View File

@@ -28,7 +28,8 @@ on `Task.current_task` (the main Task). The dictionary contains the `dist.rank`
```python
Task.current_task().upload_artifact(
'temp {:02d}'.format(dist.get_rank()), artifact_object={'worker_rank': dist.get_rank()}
name='temp {:02d}'.format(dist.get_rank()),
artifact_object={'worker_rank': dist.get_rank()}
)
```
@@ -44,7 +45,10 @@ same title (`loss`), but a different series name (containing the subprocess' `ra
```python
Task.current_task().get_logger().report_scalar(
'loss', 'worker {:02d}'.format(dist.get_rank()), value=loss.item(), iteration=i
title='loss',
series='worker {:02d}'.format(dist.get_rank()),
value=loss.item(),
iteration=i
)
```