Skip to content

Commit

Permalink
hotfix
Browse files Browse the repository at this point in the history
  • Loading branch information
@Guitaricet
Guitaricet committed Jan 4, 2024
1 parent 4dab622 commit 176f376
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchrun_main.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
import torch
import torch.nn as nn
import torch.utils.data
import torch.distributed as dist as dist
import torch.distributed as dist
from torch.distributed.optim import ZeroRedundancyOptimizer
from torch.distributed.fsdp import (
FullyShardedDataParallel as FSDP,
Expand Down

0 comments on commit 176f376

Please sign in to comment.