Unverified Commit bf1c79a6 authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Add license header to NAS (#1788)

parent 6b9ccd33
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import numpy as np import numpy as np
import torch import torch
from torchvision import transforms from torchvision import transforms
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import torch import torch
import torch.nn as nn import torch.nn as nn
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import torch import torch
import torch.nn as nn import torch.nn as nn
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import logging import logging
import time import time
from argparse import ArgumentParser from argparse import ArgumentParser
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import logging import logging
import time import time
from argparse import ArgumentParser from argparse import ArgumentParser
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
def accuracy(output, target, topk=(1,)): def accuracy(output, target, topk=(1,)):
""" Computes the precision@k for the specified values of k """ """ Computes the precision@k for the specified values of k """
maxk = max(topk) maxk = max(topk)
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from torchvision import transforms from torchvision import transforms
from torchvision.datasets import CIFAR10 from torchvision.datasets import CIFAR10
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import torch.nn as nn import torch.nn as nn
from nni.nas.pytorch import mutables from nni.nas.pytorch import mutables
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import torch import torch
import torch.nn as nn import torch.nn as nn
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import logging import logging
import time import time
from argparse import ArgumentParser from argparse import ArgumentParser
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import torch import torch
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import logging import logging
import torch.nn as nn import torch.nn as nn
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import logging import logging
import os import os
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from .mutator import DartsMutator from .mutator import DartsMutator
from .trainer import DartsTrainer from .trainer import DartsTrainer
\ No newline at end of file
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import logging import logging
import torch import torch
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import copy import copy
import logging import logging
......
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from .mutator import EnasMutator from .mutator import EnasMutator
from .trainer import EnasTrainer from .trainer import EnasTrainer
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment