Skip to content

Latest commit

 

History

History
26 lines (13 loc) · 833 Bytes

limitations.rst

File metadata and controls

26 lines (13 loc) · 833 Bytes
Raw

Limitations

Although a RaNNCModel is designed to work like torch.nn.Module, it has the following limitations.

Control constructs are ignored

RaNNC uses a computation graph produced by PyTorch's tracing function. As explained in PyTorch's document, the tracing function does not record control constructs including conditional branches and loops.

Arguments and return values

Arguments and outputs of a RaNNCModel must satisfy the following conditions.

  • Arguments must be (mini-)batches tensors, whose first dimension corresponds to samples in a mini-batch.
  • Keyword arguments are not allowed.
  • Outputs must be (mini-)batches tensors, or a loss value (scalar tensor).