diff options
| -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())) |