import dist · e0bd9fc7ce - wilds-ijepa - Gitea: Git with a cup of tea

YannAhlgrim/wilds-ijepa

import dist

This commit is contained in:

YannAhlgrim

2026-05-16 12:47:41 +02:00

parent 79fc9564ea

commit e0bd9fc7ce

1 changed files with 1 additions and 0 deletions

									
										src/train.py
									
		+1
		
												View File
												
					@@ -25,6 +25,7 @@ import yaml

					import numpy as np

					import numpy as np

					import torch

					import torch

					import torch.distributed as dist

					import torch.multiprocessing as mp

					import torch.multiprocessing as mp

					import torch.nn.functional as F

					import torch.nn.functional as F

					from torch.nn.parallel import DistributedDataParallel

					from torch.nn.parallel import DistributedDataParallel