pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Clark Kang 6d64bc3990 [data foundation][vizard] Prevent checking the device type of numpy object in Tensorboard logger (#162888 )

Summary:
The check is introduced in D82262053
- `scalar_value` could be a numpy object
  - Move the check of `device.type` into `make_np` method where it happens only when it's a `torch.Tensor`.

Test Plan:
```
vizard launch -j 1x8 --launch=flow --config-path=pkg://vizard_projects.image_classification.configs --config-name=resnet50 ++flow.secure_group=ml_sensors ++flow.entitlement=ai_frameworks_pnb ++max_train_steps_per_epoch=10 ++max_epochs=5 ++log_every_n_steps=10 ++profiler=null ++max_eval_steps_per_epoch=10
```

Rollback Plan:

Differential Revision: D82383428

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162888
Approved by: https://github.com/xush6528

2025-09-14 08:09:08 +00:00

__init__.py

Define __all__ for torch.utils.tensorboard (#147550 )

2025-02-28 23:06:11 +00:00

_convert_np.py

[data foundation][vizard] Prevent checking the device type of numpy object in Tensorboard logger (#162888 )

2025-09-14 08:09:08 +00:00

_embedding.py

Flip default value for mypy disallow_untyped_defs [10/11] (#127847 )

2024-06-08 18:50:06 +00:00

_onnx_graph.py

[BE]: Apply PERF401 autofixes from ruff (#140980 )