HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.0.29 DDP-cls and default arg fixes (#813)

Browse source
This commit is contained in:
Glenn Jocher 2023-02-06 02:30:03 +04:00 • committed by GitHub
parent 21ae321bc2
commit 7a7c8dc7b7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 38 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/engine/trainer.py
View file

@ -457,7 +457,7 @@ class BaseTrainer:
def get_validator(self):
raise NotImplementedError("get_validator function not implemented in trainer")
def get_dataloader(self, dataset_path, batch_size=16, rank=0):
def get_dataloader(self, dataset_path, batch_size=16, rank=0, mode="train"):
"""
Returns dataloader derived from torch.data.Dataloader.
"""