mixmo.utils.torchsummary.summary

mixmo.utils.torchsummary.summary(model, input_size, batch_size=2, input_initializer=<built-in method rand of type object>, list_dtype=None, device='cuda')

Model printing function