Terrace

Terrace is a library for writing concise and maintainable PyTorch code. There are two main features Terrace introduces:

  • Modules allow concise structuring of PyTorch models entirely in the model’s forward method – no more constantly switching between the __init__ and forward methods when you make changes.

  • Batches enable making nice object-oriented code compatible with PyTorch dataloaders. No more passing 10 tensors as arguments to your functions!

To get started with Terrace, simply install via pip.

pip install terrace