Name	Name	Last commit message	Last commit date
parent directory ..
config	config
data	data
README.md	README.md
__init__.py	__init__.py
custom_types.py	custom_types.py
dataset.py	dataset.py
eval.py	eval.py
model.py	model.py
run.py	run.py

Name

Last commit message

Last commit date

Example

lightning を使った ACT の学習・テストコード. rye を使うと便利です.

使い方

tonyzhaozh/ACT の Simulated experiments を実行し, シミュレーションデータを収集
We use sim_transfer_cube_scripted task in the examples below. Another option is sim_insertion_scripted. To generated 50 episodes of scripted data, run:
```
python3 record_sim_episodes.py --task_name sim_transfer_cube_scripted --dataset_dir data --num_episodes 50
```
収集したデータ (.hdf5) を examples/data ディレクトリに配置
依存関係のインストール

｀dev-dependencies` に含まれるパッケージもインストールする必要がある
```
rye sync --no-lock
```

学習

lightningCLI を使って学習を行う

cd example
python run.py fit --config config/sim_insertion_scripted.yaml

テスト

wandb にログが記録される

python run.py test --config config/test_sim_insertion_scripted.yaml