Grape / config.json
Yasssh2123's picture
Update config.json
3ad8e14 verified
{
"input_size": 640,
"task": "instance-segmentation",
"ultralytics_version": "8.0.20",
"model_type": "yolov8",
"pipeline_tag": "mask-generation",
"tags": [
"image-segmentation"
],
"model": "yolov8-seg",
"class_set": [
{
"name": "Grape",
"id": 0
}
],
"labels": [
{
"name": "Grape",
"attributes": []
}
],
"id2label": {
"0": "Grape"
},
"hyperparameters": {
"epochs": 100,
"imgsz": 640,
"batch_size": 16,
"learning_rate": 0.01,
"momentum": 0.9,
"weight_decay": 0.0005,
"optimizer": "SGD",
"lr_scheduler": "StepLR",
"step_size": 10,
"gamma": 0.1
},
"architectures": [
"YOLOv8"
],
"model_name_or_path": "Yasssh2123/Grape",
"use_cache": false,
"do_sample": false,
"early_stopping": true,
"fp16": false,
"num_beams": 1,
"use_legacy_prediction_head": false,
"license": "agpl-3.0",
"base_model": [
"Ultralytics/YOLOv8",
"raidavid/yolov8n-seg"
]
}