InHeadConv1D.compile

InHeadConv1D.compile(*args, **kwargs)[source]

Compile this Module’s forward using torch.compile().

This Module’s __call__ method is compiled and all arguments are passed as-is to torch.compile().

See torch.compile() for details on the arguments for this function.