Skip to content

Commit

Permalink
Fix isort errors
Browse files Browse the repository at this point in the history
  • Loading branch information
fsschneider committed Jan 24, 2022
1 parent d253cbb commit c474f7a
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
1 change: 1 addition & 0 deletions sample_submissions/imagenet/imagenet_jax/submission.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
from jax import lax
import jax.numpy as jnp
import optax

from algorithmic_efficiency import spec


Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
"""Training algorithm track submission functions for LibriSpeech."""
from typing import Iterator, List, Tuple

from algorithmic_efficiency import spec
import torch

from algorithmic_efficiency import spec

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
ctc_loss = torch.nn.CTCLoss(blank=0, reduction="none")

Expand Down

0 comments on commit c474f7a

Please sign in to comment.