mirror of
https://github.com/wassname/iris_bigvae.git
synced 2026-09-09 11:24:31 +08:00
29 lines
1.1 KiB
JSON
29 lines
1.1 KiB
JSON
{
|
|
// Use IntelliSense to learn about possible attributes.
|
|
// Hover to view descriptions of existing attributes.
|
|
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
|
"version": "0.2.0",
|
|
"configurations": [
|
|
{
|
|
"name": "main",
|
|
"type": "python",
|
|
"request": "launch",
|
|
"program": "${workspaceFolder}/src/main.py",
|
|
"console": "integratedTerminal",
|
|
"justMyCode": false,
|
|
"autoReload": {"enable": true,},
|
|
"env": {"WANDB_MODE":"disabled"},
|
|
"args": [
|
|
"env.train.id=BreakoutNoFrameskip-v4",
|
|
// # make it start early
|
|
"training.tokenizer.start_after_epochs=1",
|
|
"training.world_model.start_after_epochs=2",
|
|
"training.actor_critic.start_after_epochs=3",
|
|
"training.tokenizer.steps_per_epoch=40",
|
|
"training.world_model.steps_per_epoch=40",
|
|
"training.actor_critic.steps_per_epoch=40",
|
|
]
|
|
}
|
|
]
|
|
}
|