utils.py

import os
import torch
import random
import numpy as np


def save_checkpoint(state, filepath='./weights',
                    filename='latest.pth'):
    os.makedirs(filepath, exist_ok=True)
    save_path = os.path.join(filepath, filename)
    torch.save(state, save_path)


def setup_seed(seed):
    torch.manual_seed(seed)
    torch.cuda.manual_seed_all(seed)
    np.random.seed(seed)
    random.seed(seed)
    torch.backends.cudnn.deterministic = True

class AverageMeter(object):
    """Computes and stores the average and current value"""
    def __init__(self):
        self.reset()

    def reset(self):
        self.val = 0
        self.avg = 0
        self.sum = 0
        self.count = 0

    def update(self, val, n=1):
        self.val = val
        self.sum += val * n
        self.count += n
        self.avg = self.sum / self.count