Skip to content

Commit

Permalink
change delimiter
Browse files Browse the repository at this point in the history
  • Loading branch information
tlpss committed Sep 29, 2023
1 parent 58f5937 commit 64b3bdd
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion keypoint_detection/tasks/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ def add_system_args(parent_parser: ArgumentParser) -> ArgumentParser:
parser.add_argument(
"--keypoint_channel_configuration",
type=str,
help="A list of the semantic keypoints that you want to learn in each channel. These semantic categories must be defined in the COCO dataset. Seperate the channels with a ; and the categories within a channel with a =",
help="A list of the semantic keypoints that you want to learn in each channel. These semantic categories must be defined in the COCO dataset. Seperate the channels with a : and the categories within a channel with a =",
)

parser.add_argument(
Expand Down
2 changes: 1 addition & 1 deletion test/integration_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,6 @@

# make sure to remove all trailing spaces from the command, as this would result in an error when using bash.
python keypoint_detection/tasks/train.py \
--keypoint_channel_configuration "box_corner0= box_corner1 = box_corner2= box_corner3; flap_corner0 ; flap_corner2" \
--keypoint_channel_configuration "box_corner0= box_corner1 = box_corner2= box_corner3: flap_corner0:flap_corner2" \
--json_dataset_path "test/test_dataset/coco_dataset.json" --json_validation_dataset_path "test/test_dataset/coco_dataset.json" --batch_size 2 --wandb_project "keypoint-detector-integration-test" \
--max_epochs 50 --early_stopping_relative_threshold -1.0 --log_every_n_steps 1 --accelerator="gpu" --devices 1 --precision 16 --augment_train

0 comments on commit 64b3bdd

Please sign in to comment.