Skip to content

Instantly share code, notes, and snippets.

@TronicLT
TronicLT / Pytorchtrain.py
Last active June 28, 2023 04:02
Pytorch train
# coding=utf-8
import types
import torch
import torch.nn as nn
from torch.utils import data
from saga.data.datasets import ArrayDataset
from saga.utils.callbacks import Callback, History, ProgressBar, Callbacks
from saga.utils.general_utils import check_attribute