Skip to content

Commit

Permalink
linter 2
Browse files Browse the repository at this point in the history
  • Loading branch information
Pherkel committed Sep 18, 2023
1 parent d9421ca commit a4049cb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions swr2_asr/utils/visualization.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()))
Expand Down

0 comments on commit a4049cb

Please sign in to comment.