add use_strict_trace to tensorboard add_graph method (#63120)
authorEdward Wang (EcoF) <hack@fb.com>
Thu, 12 Aug 2021 19:10:50 +0000 (12:10 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 12 Aug 2021 19:12:12 +0000 (12:12 -0700)
commit96fb1a56ea894b098df48b45adc053b8bf7f9a37
tree1a47380b242a7d69433d769e811d83bef90c4439
parent1022443168b5fad55bbd03d087abf574c9d2e9df
add use_strict_trace to tensorboard add_graph method (#63120)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63120

FAIM returns dictionaries as the model output, which throws an error when trying to trace using add_graph. Pass in `strict` to the tracer to make this user configurable.

User post: https://fb.workplace.com/groups/pytorchLightning/permalink/1510194972650369/?comment_id=1510252919311241&reply_comment_id=1510281112641755

Test Plan: unit test

Reviewed By: Reubend

Differential Revision: D30265890

fbshipit-source-id: 58b25d9500b875a29a664aa9ef4c1e7f13631fa1
test/test_tensorboard.py
torch/utils/tensorboard/_pytorch_graph.py
torch/utils/tensorboard/writer.py