Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Mar 15, 2024
1 parent 67b234e commit 397cb99
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -137,7 +137,6 @@ def main(args):
data_list_file_path=json_path, is_segmentation=False, data_list_key="validation", base_dir=data_root
)


print("Total training data are {} and validation data are {}".format(len(train_list), len(val_list)))

train_dataset = CacheDataset(data=train_list, transform=train_transforms, cache_rate=1.0, num_workers=4)
Expand Down Expand Up @@ -186,7 +185,7 @@ def main(args):
optimizer = torch.optim.Adam(model.parameters(), lr=args.base_lr)

model = torch.nn.parallel.DistributedDataParallel(
model, device_ids=[int(os.environ['LOCAL_RANK'])], broadcast_buffers=False, find_unused_parameters=True
model, device_ids=[int(os.environ["LOCAL_RANK"])], broadcast_buffers=False, find_unused_parameters=True
)
model_without_ddp = model.module

Expand Down

0 comments on commit 397cb99

Please sign in to comment.