Skip to main content

Module space

Module space 

Source
Expand description

Neural architecture search space definitions.

Defines ArchSearchSpace, Architecture, and LayerSpec — the primitives that describe which architectures are valid candidates during NAS.

Structs§

ArchSearchSpace
Defines the search space over neural architectures.
Architecture
A concrete neural architecture represented as an ordered sequence of layers.
LayerSpec
Specification for a single layer in a neural architecture.