Source code for power_cogs.base.base_torch_model

import torch
from torchsummaryX import summary


[docs]class BaseTorchModel(torch.nn.Module): def __init__(self): super(BaseTorchModel, self).__init__()
[docs] def summary(self, input_shape): summary(self, input_shape)