from super_gradients.training import models
from torchinfo import summary
import torch