Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ hyperparameters:
- 2.0
probability: 0.5
color_jitter:
enable: true
enable: false
brightness:
- 0.875
- 1.125
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,3 +16,7 @@ hyperparameters:
learning_rate: 0.0001
input_size_width: 640
input_size_height: 640
dataset_preparation:
augmentation:
iou_random_crop:
enable: false
Original file line number Diff line number Diff line change
Expand Up @@ -16,3 +16,7 @@ hyperparameters:
learning_rate: 0.0001
input_size_width: 640
input_size_height: 640
dataset_preparation:
augmentation:
iou_random_crop:
enable: false
Original file line number Diff line number Diff line change
Expand Up @@ -15,3 +15,7 @@ hyperparameters:
learning_rate: 0.0001
input_size_width: 640
input_size_height: 640
dataset_preparation:
augmentation:
iou_random_crop:
enable: false
2 changes: 1 addition & 1 deletion interactive_ai/workflows/train/trainer/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ dependencies = [
"otx",
"geti-kafka-tools",
"geti-telemetry-tools[telemetry,kafka]",
"setuptools==69.5.1",
"setuptools==78.1.1",
"av<16",
]

Expand Down
15 changes: 7 additions & 8 deletions interactive_ai/workflows/train/trainer/uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 0 additions & 1 deletion interactive_ai/workflows/train/trainer/xpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,6 @@ COPY --link download_pretrained_weights.py download_pretrained_weights.py
RUN --mount=type=cache,target=/root/.cache/uv uv sync --frozen --no-dev --no-editable
RUN uv pip uninstall torch torchvision
RUN uv pip install torch==2.8.0 torchvision==0.23.0 --default-index https://download.pytorch.org/whl/xpu
RUN uv pip install setuptools==69.5.1

# Place executables in the environment at the front of the path
ENV PATH="/interactive_ai/workflows/train/trainer/.venv/bin:/interactive_ai/workflows/train/trainer:$PATH"
Expand Down
Loading