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