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.optim as optim
from os import *
from os import system, name
def clear():
if os.name == 'nt':
_ = os.system('cls')
if name == 'nt':
_ = system('cls')
else:
_ = os.system('clear')
_ = system('clear')
def train(device, net, epochs, trainloader, optimizer, criterion):