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)