Config-only Mode
If you need Component
objects only for configurations,
you can use no_parameters
context to suppress the creation of any
parameters and only set the config
attribute.
import torchmix
from torchmix import nn
with torchmix.no_parameters():
nn.Linear(1000000, 1000000) # Fast! ⚡ ️