| import random |
| import torch |
| import os |
|
|
|
|
| def seed_everything(seed_value=42): |
| os.environ['PYTHONHASHSEED'] = str(seed_value) |
| random.seed(seed_value) |
| torch.manual_seed(seed_value) |
| torch.cuda.manual_seed_all(seed_value) |
|
|
| torch.backends.cudnn.enabled = True |
| torch.backends.cudnn.deterministic = True |
| torch.backends.cudnn.benchmark = False |
|
|
|
|
| def initialize(args, init_wandb: bool): |
| seed_everything(args.seed) |
|
|
| if init_wandb: |
| import wandb |
| tags = args.framework, args.language |
| wandb.init(name=f"{args.framework}_{args.language}_{args.graph_mode}_{args.name}", config=args, project="sentiment_graphs", tags=list(tags)) |
| print("Connection to Weights & Biases initialized.", flush=True) |
|
|
