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

@@ -3,11 +3,9 @@
########################################################
# DARTS: Differentiable Architecture Search, ICLR 2019 #
########################################################
import os, sys, time, glob, random, argparse
import numpy as np
import sys, time, random, argparse
from copy import deepcopy
import torch
import torch.nn as nn
from pathlib import Path
lib_dir = (Path(__file__).parent / '..' / '..' / 'lib').resolve()
if str(lib_dir) not in sys.path: sys.path.insert(0, str(lib_dir))