fix: tensor dimensions

fix execution error
This commit is contained in:
Henri Bourcereau 2025-06-29 11:30:34 +02:00
parent 6a7b1cbebc
commit b98a135749
2 changed files with 5 additions and 4 deletions

View file

@ -19,5 +19,6 @@ pythonlib:
pip install --no-deps --force-reinstall --prefix .devenv/state/venv target/wheels/*.whl
trainbot:
#python ./store/python/trainModel.py
# cargo run --bin=train_dqn
cargo run --bin=train_burn_rl
# cargo run --bin=train_dqn # ok
# cargo run --bin=train_burn_rl # doesn't save model
cargo run --bin=train_dqn_full