Commit cd38e156 authored by Mayumi Ohta's avatar Mayumi Ohta
Browse files

set seed in torch.cuda

parent aed3fa4c

joeynmt/helpers.py

+3 −0
Original line number Diff line number Diff line
@@ -126,6 +126,9 @@ def set_seed(seed: int) -> None: 
    torch.manual_seed(seed)

    np.random.seed(seed)

    random.seed(seed)

    if torch.cuda.is_available() and torch.cuda.device_count() > 0:

        torch.backends.cudnn.deterministic = True

        torch.cuda.manual_seed_all(seed)





def log_data_info(train_data: Dataset, valid_data: Dataset, test_data: Dataset,