pytholic/streamlit-image-classification-demo

1from dataclasses import dataclass2 3 4@dataclass5class Args:6    """7    Training arguments.8    """9 10    # Learning rate for the optimizer11    learning_rate: float = 1e-312    # Training batch size13    batch_size: int = 6414    # Total numebr of classes15    num_classes: int = 1016    # Maximum number of training epochs17    max_epochs: int = 10018    # Input shape19    input_shape: tuple = (3, 224, 224)20    # Use pretrained weights21    # Can be "IMAGENET1K_V1", "IMAGENET1K_V2", "DEFAULT"22    # CHec more at https://pytorch.org/vision/stable/models.html23    weights: str = None24