pytorch-models | tool for managing and modularizing PyTorch modules | Machine Learning library

models
  ∟ __init__.py
  ∟ mlp.py

import torchmodels
torchmodels.register_packages(models)

import torchmodels

class MLP(torchmodels.Module):

    name = "some-mlp"

    def __init__(self, input_dim, output_dim, *, 
                 hidden_dim=300,  

type: multiplicative-attention
vargs:
  hidden_dim: 200

import torch
import torchmodels
from torchmodels.modules import attention

model_cls = torchmodels.create_model_cls(attention, model_path="att.yml")
model = model_cls(
    qry_dim=200,
    val_