mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

refactored

Browse Source
This commit is contained in:
Bala-Vignesh-Reddy 2025-03-22 23:29:38 +05:30
commit c4ec77fa98
3 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
classify/train.py
View File

@ -28,10 +28,8 @@ import torch.distributed as dist
import torch.hub as hub import torch.hub as hub
import torch.optim.lr_scheduler as lr_scheduler import torch.optim.lr_scheduler as lr_scheduler
import torchvision import torchvision
from torch.cuda import amp
from tqdm import tqdm from tqdm import tqdm
FILE = Path(__file__).resolve() FILE = Path(__file__).resolve()
ROOT = FILE.parents[1] # YOLOv5 root directory ROOT = FILE.parents[1] # YOLOv5 root directory
if str(ROOT) not in sys.path: if str(ROOT) not in sys.path:

1
segment/train.py
View File

@ -90,7 +90,6 @@ from utils.torch_utils import (
torch_distributed_zero_first, torch_distributed_zero_first,
) )
LOCAL_RANK = int(os.getenv("LOCAL_RANK", -1)) # https://pytorch.org/docs/stable/elastic/run.html LOCAL_RANK = int(os.getenv("LOCAL_RANK", -1)) # https://pytorch.org/docs/stable/elastic/run.html
RANK = int(os.getenv("RANK", -1)) RANK = int(os.getenv("RANK", -1))
WORLD_SIZE = int(os.getenv("WORLD_SIZE", 1)) WORLD_SIZE = int(os.getenv("WORLD_SIZE", 1))

1
train.py
View File

@ -95,7 +95,6 @@ from utils.torch_utils import (
torch_distributed_zero_first, torch_distributed_zero_first,
) )
LOCAL_RANK = int(os.getenv("LOCAL_RANK", -1)) # https://pytorch.org/docs/stable/elastic/run.html LOCAL_RANK = int(os.getenv("LOCAL_RANK", -1)) # https://pytorch.org/docs/stable/elastic/run.html
RANK = int(os.getenv("RANK", -1)) RANK = int(os.getenv("RANK", -1))
WORLD_SIZE = int(os.getenv("WORLD_SIZE", 1)) WORLD_SIZE = int(os.getenv("WORLD_SIZE", 1))