Skip to content

Commit 80c65ff

Browse files
SeongooSeongoo
authored
Merge pull request #40 from ROBOTIS-GIT/feature-training-manager
Add the Training Module
2 parents fb15b35 + d4754ec commit 80c65ff

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

45 files changed

+3086
-146
lines changed

‎physical_ai_interfaces/CHANGELOG.rst‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,11 @@
22
Changelog for package physical_ai_interfaces
33
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
44

5+
0.6.0 (2025-07-23)
6+
------------------
7+
* Added training-related service types and message types
8+
* Contributors: Kiwoong Park, Seongwoo Kim
9+
510
0.5.13 (2025-07-21)
611
------------------
712
* None

‎physical_ai_interfaces/CMakeLists.txt‎

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,20 @@ find_package(rosidl_default_generators REQUIRED)
77
rosidl_generate_interfaces(${PROJECT_NAME}
88
"msg/TaskInfo.msg"
99
"msg/TaskStatus.msg"
10+
"msg/TrainingInfo.msg"
11+
"msg/TrainingStatus.msg"
12+
"srv/GetDatasetList.srv"
1013
"srv/GetHFUser.srv"
1114
"srv/GetImageTopicList.srv"
15+
"srv/GetModelWeightList.srv"
1216
"srv/GetPolicyList.srv"
1317
"srv/GetRobotTypeList.srv"
1418
"srv/GetSavedPolicyList.srv"
19+
"srv/GetUserList.srv"
1520
"srv/SendCommand.srv"
16-
"srv/SetRobotType.srv"
21+
"srv/SendTrainingCommand.srv"
1722
"srv/SetHFUser.srv"
23+
"srv/SetRobotType.srv"
1824
)
1925

2026
install(

‎physical_ai_interfaces/msg/TrainingInfo.msg‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
# Must be given
2+
string dataset
3+
string policy_type
4+
string output_folder_name
5+
string policy_device
6+
7+
# Options
8+
uint32 seed
9+
uint8 num_workers
10+
uint16 batch_size
11+
uint32 steps
12+
uint32 eval_freq
13+
uint32 log_freq
14+
uint32 save_freq

‎physical_ai_interfaces/msg/TrainingStatus.msg‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
########################################
2+
# Constants
3+
########################################
4+
5+
TrainingInfo training_info
6+
uint32 current_step
7+
bool is_training
8+
string error
9+
# diagnostic_msgs/KeyValue[] metrics # To be determined

‎physical_ai_interfaces/package.xml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
<?xml-model href="http://download.ros.org/schema/package_format3.xsd" schematypens="http://www.w3.org/2001/XMLSchema"?>
33
<package format="3">
44
<name>physical_ai_interfaces</name>
5-
<version>0.5.13</version>
5+
<version>0.6.0</version>
66
<description>
77
ROS 2 interfaces for Physical AI tools
88
</description>

‎physical_ai_interfaces/srv/GetDatasetList.srv‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
string user_id
2+
---
3+
string[] dataset_list
4+
bool success
5+
string message

‎physical_ai_interfaces/srv/GetModelWeightList.srv‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
---
2+
string[] model_weight_list
3+
bool success
4+
string message

‎physical_ai_interfaces/srv/GetPolicyList.srv‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
---
22
string[] policy_list
3+
string[] device_list
34
bool success
45
string message

‎physical_ai_interfaces/srv/GetUserList.srv‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
---
2+
string[] user_list
3+
bool success
4+
string message

‎physical_ai_interfaces/srv/SendTrainingCommand.srv‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
########################################
2+
# Constants
3+
########################################
4+
# command
5+
uint8 NONE = 0
6+
uint8 START = 1
7+
uint8 FINISH = 2
8+
9+
uint8 command
10+
TrainingInfo training_info
11+
---
12+
bool success
13+
string message

0 commit comments

Comments
 (0)