diff options
| author | Pherkel | 2023-09-18 21:52:33 +0200 |
|---|---|---|
| committer | Pherkel | 2023-09-18 21:52:33 +0200 |
| commit | d9421cafa088babdf6b84fa18af28ba6671addce (patch) | |
| tree | dc92662ff0bc039b898ee4275dfc456b702d4c59 /swr2_asr/utils | |
| parent | eaec9a8a32fc791c9ea676843dd7cac6a2b48d9a (diff) | |
linter changes
Diffstat (limited to 'swr2_asr/utils')
| -rw-r--r-- | swr2_asr/utils/visualization.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/swr2_asr/utils/visualization.py b/swr2_asr/utils/visualization.py index b288c5a..23956fd 100644 --- a/swr2_asr/utils/visualization.py +++ b/swr2_asr/utils/visualization.py @@ -14,7 +14,9 @@ def plot(path): epoch = 5 while True: try: - current_state = torch.load(path + str(epoch), map_location=torch.device("cpu")) + current_state = torch.load( + path + str(epoch), map_location=torch.device("cpu") + ) # pylint: disable=no-member except FileNotFoundError: break train_losses.append((epoch, current_state["train_loss"].item())) |