From c051ffecc6615b5984b4ed9bda71ab3fe75cc902 Mon Sep 17 00:00:00 2001 From: "Lin, Fanli" Date: Fri, 7 Feb 2025 23:11:53 -0800 Subject: [PATCH 1/2] replace set_seed --- examples/cv_example.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/examples/cv_example.py b/examples/cv_example.py index 87d1575e08e..65c3798cb70 100644 --- a/examples/cv_example.py +++ b/examples/cv_example.py @@ -22,6 +22,7 @@ from torch.optim.lr_scheduler import OneCycleLR from torch.utils.data import DataLoader, Dataset from torchvision.transforms import Compose, RandomResizedCrop, Resize, ToTensor +from transformers import set_seed from accelerate import Accelerator @@ -93,10 +94,7 @@ def training_function(config, args): label_to_id = {lbl: i for i, lbl in enumerate(id_to_label)} # Set the seed before splitting the data. - np.random.seed(seed) - torch.manual_seed(seed) - torch.cuda.manual_seed_all(seed) - + set_seed(seed) # Split our filenames between train and validation random_perm = np.random.permutation(len(file_names)) cut = int(0.8 * len(file_names)) From 57ef807bada2e88188bb27826efc93b1a2d324e3 Mon Sep 17 00:00:00 2001 From: "Lin, Fanli" Date: Mon, 10 Feb 2025 21:12:45 -0800 Subject: [PATCH 2/2] update import --- examples/cv_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/cv_example.py b/examples/cv_example.py index 65c3798cb70..f7e268ee661 100644 --- a/examples/cv_example.py +++ b/examples/cv_example.py @@ -22,9 +22,9 @@ from torch.optim.lr_scheduler import OneCycleLR from torch.utils.data import DataLoader, Dataset from torchvision.transforms import Compose, RandomResizedCrop, Resize, ToTensor -from transformers import set_seed from accelerate import Accelerator +from accelerate.utils import set_seed ########################################################################