Update Weight Watcher in utils

This commit is contained in:
D-X-Y
2020-03-11 00:44:39 -07:00
parent 2964bda731
commit 1fcde3e8ac
5 changed files with 346 additions and 12 deletions

View File

@@ -1,11 +1,10 @@
# This file is for experimental usage
import os, sys, torch, random
import torch, random
import numpy as np
from copy import deepcopy
from tqdm import tqdm
import torch.nn as nn
from utils import obtain_accuracy
# from utils import obtain_accuracy
from models import CellStructure
from log_utils import time_string