Commit Graph

67 Commits

Author SHA1 Message Date
gouhanke
7d39933a5b feat: 缓存worker内的句柄 2026-03-04 10:49:41 +08:00
gouhanke
8bcad5844e fix: 修复VLA设备与损失计算逻辑,并优化Transformer默认训练参数 2026-03-03 17:56:12 +08:00
gouhanke
cdb887c9bf feat: 添加transformer头 2026-02-28 19:07:27 +08:00
gouhanke
abb4f501e3 chore: 删除unet里的local_cond(未使用) 2026-02-28 10:42:16 +08:00
gouhanke
1d33db0ef0 chore: 缩小物块的大小 2026-02-27 18:23:30 +08:00
gouhanke
f27e397f98 chore: 修改了采数时的一些参数 2026-02-26 17:09:40 +08:00
gouhanke
4e0add4e1d debug: 修复episode首帧图像不正确的问题;修复前2个episode帧重复的问题 2026-02-26 16:17:54 +08:00
gouhanke
40c40695dd chore: 添加测试文件
- check_all_episodes.py:检查各个episode是否有重复帧。
- check_specific_frames.py:检查前几帧是否位于正确初始位置。
- generate_dataset_videos.py:根据hdf5生成视频
2026-02-26 13:59:47 +08:00
gouhanke
3deeffb9fe chore:改变了一些参数配置 2026-02-26 13:56:03 +08:00
gouhanke
0b05c01024 feat: 推理时输出action 2026-02-12 19:54:11 +08:00
gouhanke
926a78eb66 feat: 添加finetune 2026-02-12 19:31:44 +08:00
gouhanke
efbe4b6ac9 Revert "Merge branch 'dev' of gitlab.com:leeeezd0016-group/gouhanke-vla into dev"
This reverts commit acb1467473, reversing
changes made to 624b926e33.
2026-02-12 18:31:56 +08:00
gouhanke
acb1467473 Merge branch 'dev' of gitlab.com:leeeezd0016-group/gouhanke-vla into dev 2026-02-12 18:08:27 +08:00
gouhanke
624b926e33 debug: 添加推理时缩放,加大采数以及推理时物块的放置范围 2026-02-12 17:14:23 +08:00
gouhanke
926d8cf894 chore: 加载时将图像缩放到224*224, resnet禁用crop 2026-02-12 15:02:18 +08:00
gouhanke
116ba13fb9 chore: 验证归一化是否有效 2026-02-12 13:01:13 +08:00
gouhanke
37a47ac2dd debug: 保存stats到ckpt 2026-02-12 13:00:43 +08:00
gouhanke
ab971b3f96 debug: 归一化 2026-02-12 12:23:34 +08:00
gouhanke
83cd55e67b 添加pad_loss 2026-02-11 20:33:26 +08:00
gouhanke
eeb07cad15 feat: 冻结resnet 2026-02-11 20:11:25 +08:00
gouhanke
83d11ab640 Merge branch 'dev' of gitlab.com:leeeezd0016-group/gouhanke-vla into dev 2026-02-11 17:20:21 +08:00
JiajunLI
aba8779671 更改默认参数 2026-02-11 17:14:32 +08:00
gouhanke
b42c1c68fd debug: 将归一化放在GPU上 2026-02-11 17:13:55 +08:00
gouhanke
320369ffb8 debug: 归一化图像到[0, 1] 2026-02-11 16:47:39 +08:00
gouhanke
130d4bb3c5 refactor:大重构 2026-02-11 15:53:55 +08:00
gouhanke
1e95d40bf9 debug 2026-02-10 15:56:05 +08:00
gouhanke
3c27d6d793 refactor: 重构resnet 2026-02-10 15:26:10 +08:00
gouhanke
88b9c10a75 refactor(dataset): 重新创建robotdataset最小实现
- 内部实现__getitem__参数,可以通过滑动窗口进行采样
-
2026-02-10 10:26:19 +08:00
gouhanke
ac870f6110 chore: 计算推理频率 2026-02-09 15:39:22 +08:00
gouhanke
8b700b6d99 暂存 2026-02-09 14:41:35 +08:00
gouhanke
f833c6d9f1 添加readme文件 2026-02-07 09:57:59 +08:00
gouhanke
4332530a5f feat(train): 添加warmup学习率调度器 2026-02-06 22:54:34 +08:00
gouhanke
456056347f debug: 固定验证集上的随机噪声,修复resnet在训练时bn层会切换到train的问题 2026-02-06 21:31:19 +08:00
gouhanke
05f3cc1e47 chore: 删除detr和gr00t 2026-02-06 20:21:01 +08:00
gouhanke
a6fcb88203 chore: 删除多余文件 2026-02-06 20:19:11 +08:00
gouhanke
3d0c2ec5b1 feat(train): 添加验证集 2026-02-06 18:00:09 +08:00
gouhanke
ea49e63eb7 feat: 注册了自定义 resolver计算长度 2026-02-06 16:08:56 +08:00
gouhanke
7a9ca06aa0 feat(dependency): 生成environment.yml文件 2026-02-06 15:40:24 +08:00
gouhanke
f006d50814 chore: 自动获取cameras的长度 2026-02-06 15:33:07 +08:00
gouhanke
f4a5c77b7c refactor: 归一化从agent解耦到训练、推理脚本中 2026-02-06 14:29:36 +08:00
gouhanke
a43a2e3d18 chore: 删除多余脚本 2026-02-06 13:45:35 +08:00
gouhanke
31419a6fc1 chore(camera): 添加front相机 2026-02-06 11:53:01 +08:00
gouhanke
66009473ad debug(inference): 添加推理阶段qpos归一化 2026-02-06 09:00:44 +08:00
gouhanke
b0a944f7aa feat(train): 跑通训练脚本 2026-02-05 14:08:43 +08:00
gouhanke
dd2749cb12 feat: 更新框架,新增数据及定义和backbone 2026-02-05 01:44:43 +08:00
gouhanke
92660562fb feat(dataset): 添加统计数据计算脚本 2026-02-05 01:44:43 +08:00
gouhanke
03f10b0c22 feat: 编写状态编码器、动作编码器 2026-02-05 01:44:43 +08:00
gouhanke
8fce9c89ef chore: 删除多余文件 2026-02-05 01:44:43 +08:00
JiajunLI
30b8ff4d7d Merge branch 'main' into 'dev'
# Conflicts:
#   README.md
2026-02-04 09:30:00 +00:00
gouhanke
3f8c3dbf5d chore(readme): 修改readme里的数据结构标准 2026-02-04 14:33:52 +08:00