from torch import nn from omegaconf import DictConfig from timm import model import pytorch_lightning as ptl Dict class VGG16(ptl.LightningModule: def