Spaces:

Hemanth0004
/

Momask-Demo-Project

Build error

App Files Files Community

Momask-Demo-Project / utils /fixseed.py

Hemanth0004

Uploaded Utils

6d426fe verified 11 months ago

Raw History Blame Contribute Delete

351 Bytes

import numpy as np
import torch
import random


def fixseed(seed):
    torch.backends.cudnn.benchmark = False
    random.seed(seed)
    np.random.seed(seed)
    torch.manual_seed(seed)


# SEED = 10
# EVALSEED = 0
# # Provoc warning: not fully functionnal yet
# # torch.set_deterministic(True)
# torch.backends.cudnn.benchmark = False
# fixseed(SEED)

	import numpy as np
	import torch
	import random


	def fixseed(seed):
	torch.backends.cudnn.benchmark = False
	random.seed(seed)
	np.random.seed(seed)
	torch.manual_seed(seed)


	# SEED = 10
	# EVALSEED = 0
	# # Provoc warning: not fully functionnal yet
	# # torch.set_deterministic(True)
	# torch.backends.cudnn.benchmark = False
	# fixseed(SEED)