YOLO Train

YOLOTrainNode

custom_nodes.GeoNodes

Inputs (15)

Name	Type	Required
dataset_path	STRING	required
model_size	COMBO	required
task	COMBO	required
epochs	INT	required
batch_size	INT	required
img_size	INT	required
device	COMBO	required
pretrained	BOOLEAN	required
learning_rate	FLOAT	required
patience	INT	required
val_split	FLOAT	required
project_name	STRING	optional
name	STRING	optional
resume	BOOLEAN	optional
checkpoint_path	STRING	optional

Outputs (3)

Name	Type
model_path	STRING
results_dir	STRING
status	STRING