Skip to content

Commit

Permalink
sort imports
Browse files Browse the repository at this point in the history
  • Loading branch information
jarlsondre committed Jan 7, 2025
1 parent 4b6705f commit 6897039
Showing 1 changed file with 0 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -20,14 +20,12 @@

import horovod.torch as hvd
import torch

import torchvision
from torch.utils.data import DataLoader
from torch.utils.data.distributed import DistributedSampler
from utils import get_parser, imagenet_dataset, train_epoch

from itwinai.loggers import EpochTimeTracker

from itwinai.torch.distributed import (
DeepSpeedStrategy,
HorovodStrategy,
Expand Down

0 comments on commit 6897039

Please sign in to comment.