18 lines
488 B
Python
18 lines
488 B
Python
#####################################################
|
|
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.01 #
|
|
#####################################################
|
|
|
|
import abc
|
|
import torch.nn as nn
|
|
|
|
|
|
class SuperModule(abc.ABCMeta, nn.Module):
|
|
"""This class equips the nn.Module class with the ability to apply AutoDL."""
|
|
|
|
def __init__(self):
|
|
super(SuperModule, self).__init__()
|
|
|
|
@abc.abstractmethod
|
|
def abstract_search_space(self):
|
|
raise NotImplementedError
|