text_html_output_1.png
import numpy as np

import os

import random

import torch

class CFG:
    seed = 46
    debug = False
    n_fold = 4
    n_epoch = 11
    height = 480
    width = 480
    model_name = 'efficientnet_b0'
    lr = 0.0001
    min_lr = 1e-06
    weight_decay = 0.0001
    T_max = 10
    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(f'Using device {CFG.device}')
def seed_torch(seed):
    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
seed_torch(CFG.seed)