admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-21 11:39:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

add multipe timesteps

Browse Source
This commit is contained in:
Pepijn
2025-08-27 16:33:53 +02:00
parent 681be962ae
commit 450be9d7d1
4 changed files with 38 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/rlearn_evaluation.ipynb
+3 -3
View File
@@ -21,7 +21,7 @@
},
{
"cell_type": "code",
"execution_count": 4,
"execution_count": 31,
"metadata": {},
"outputs": [
{
@@ -75,7 +75,7 @@
},
{
"cell_type": "code",
"execution_count": 25,
"execution_count": null,
"metadata": {},
"outputs": [
{
@@ -93,7 +93,7 @@
"source": [
"# Configuration\n",
"DATASET_REPO = \"pepijn223/rewards_bc_z3\" # Change to your dataset\n",
"MODEL_PATH = \"path/to/your/trained/model\" # Change to your model checkpoint\n",
"MODEL_PATH = \"pepijn223/rlearn_rewards_bc_z\" # Change to your model checkpoint\n",
"DEVICE = \"cuda\" if torch.cuda.is_available() else \"cpu\"\n",
"NUM_EVAL_EPISODES = 50 # Number of episodes for evaluation\n",
"\n",