Orphen2024 commited on
Commit
84964b6
·
verified ·
1 Parent(s): 5431c08

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"model_name": "resnet50", "num_labels": 7, "image_size": 224, "pretrained": true, "input_channels": 3, "output_activation": "softmax", "loss_function": "CrossEntropyLoss", "optimizer": "Adam", "learning_rate": 0.001, "transformations": {"Resize": [224, 224], "Grayscale": false, "RandomHorizontalFlip": true, "ColorJitter": {"brightness": 0.2, "contrast": 0.2, "saturation": 0.2, "hue": 0.1}, "Normalize": {"mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225]}}}