correct overlapping checkpoint evaluation
This commit is contained in:
@@ -56,6 +56,7 @@ fi
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
|
||||
DATASET_YAML="${OPERATOR_YOLO_DATASET_DIR%/}/dataset.yaml"
|
||||
NO_TRAINING_MARKER="${OPERATOR_YOLO_DATASET_DIR%/}/NO_TRAINING.json"
|
||||
SUMMARY_PATH="${TRAIN_OUTPUT_DIR%/}/${TRAIN_RUN_NAME}/training_summary.json"
|
||||
export DATASET_YAML
|
||||
export YOLO_BASE_MODEL_PATH
|
||||
@@ -76,6 +77,11 @@ if [[ ! -f "${DATASET_YAML}" ]]; then
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [[ -f "${NO_TRAINING_MARKER}" ]]; then
|
||||
echo "Training is prohibited for this evaluation-only dataset: ${NO_TRAINING_MARKER}" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [[ ! -f "${YOLO_BASE_MODEL_PATH}" ]]; then
|
||||
echo "Base model file not found: ${YOLO_BASE_MODEL_PATH}" >&2
|
||||
exit 1
|
||||
|
||||
Reference in New Issue
Block a user