This commit is contained in:
art 2019-10-29 12:25:18 -05:00
parent 2b42127971
commit 28bc2b2f21
2 changed files with 4 additions and 4 deletions

View File

@ -33,13 +33,13 @@ __all__ = [
import torch import torch
import torch.optim as optim import torch.optim as optim
from os import * from os import system, name
def clear(): def clear():
if os.name == 'nt': if name == 'nt':
_ = os.system('cls') _ = system('cls')
else: else:
_ = os.system('clear') _ = system('clear')
def train(device, net, epochs, trainloader, optimizer, criterion): def train(device, net, epochs, trainloader, optimizer, criterion):