(
self,
project: str | None = None,
entity: str | None = None,
name: str | None = None,
dir: str | None = None,
mode: Literal["online", "offline", "disabled"] | None = None,
job_type: str | None = None,
**kwargs: Any,
)
| 1184 | """ |
| 1185 | |
| 1186 | def __init__( |
| 1187 | self, |
| 1188 | project: str | None = None, |
| 1189 | entity: str | None = None, |
| 1190 | name: str | None = None, |
| 1191 | dir: str | None = None, |
| 1192 | mode: Literal["online", "offline", "disabled"] | None = None, |
| 1193 | job_type: str | None = None, |
| 1194 | **kwargs: Any, |
| 1195 | ) -> None: |
| 1196 | self.wandb = try_import( |
| 1197 | "wandb", |
| 1198 | "You want to use `wandb` which is not installed yet install it with `pip install wandb`", |
| 1199 | ) |
| 1200 | |
| 1201 | self.wandb_args = { |
| 1202 | 'project': project, |
| 1203 | 'name': name, |
| 1204 | 'entity': entity, |
| 1205 | 'dir': dir, |
| 1206 | 'mode': mode, |
| 1207 | 'job_type': job_type, |
| 1208 | } |
| 1209 | |
| 1210 | self._run = None |
| 1211 | self.wandb_args.update(**kwargs) |
| 1212 | |
| 1213 | _ = self.run |
| 1214 | |
| 1215 | def _is_write(self): |
| 1216 | return paddle.distributed.ParallelEnv().local_rank == 0 |
nothing calls this directly
no test coverage detected