Skip to content

trainNet Hydra port (?) #65

@vviggyy

Description

@vviggyy

Testing out networks with ResNet reveals that we need to add an arg parse for encoder. Currently in trainNet.py the call to make_env doesn't accept an encoder at all.

Also it would make setup faster if certain combinations of arguments should be passed in (modularly) together:

--env='MiniWorld-LRoom-v0' \ --envPackage='miniworld_vae' \ --agent='MiniworldRandomAgent' \ --actenc='ContSpeedOnehotHDMiniworld'

(e.g. when needing to switch between minigrid and miniworld you'd need to change all four of these individually, but it would be nice to have presets for these...

... enter: hydra!

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions