utility/initialize.py

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)