Spaces:

MukeshKapoor25
/

logbert_processor

Runtime error

logbert_processor / bert_pytorch /dataset /utils.py

changs

6f2ff70 over 1 year ago

538 Bytes

import random
import os
import numpy as np
import torch


def save_parameters(options, filename):
    with open(filename, "w+") as f:
        for key in options.keys():
            f.write("{}: {}\n".format(key, options[key]))

# https://gist.github.com/KirillVladimirov/005ec7f762293d2321385580d3dbe335
def seed_everything(seed=1234):
    random.seed(seed)
    os.environ['PYTHONHASHSEED'] = str(seed)
    np.random.seed(seed)
    torch.manual_seed(seed)
    # torch.cuda.manual_seed(seed)
    # torch.backends.cudnn.deterministic = True

	import random
	import os
	import numpy as np
	import torch


	def save_parameters(options, filename):
	with open(filename, "w+") as f:
	for key in options.keys():
	f.write("{}: {}\n".format(key, options[key]))

	# https://gist.github.com/KirillVladimirov/005ec7f762293d2321385580d3dbe335
	def seed_everything(seed=1234):
	random.seed(seed)
	os.environ['PYTHONHASHSEED'] = str(seed)
	np.random.seed(seed)
	torch.manual_seed(seed)
	# torch.cuda.manual_seed(seed)
	# torch.backends.cudnn.deterministic = True