Skip to content

plugins.babyai_plugin.babyai_models#

[view_source]

BabyAIACModelWrapped#

class BabyAIACModelWrapped(babyai.model.ACModel)

[view_source]

BabyAIACModelWrapped.forward_once#

 | forward_once(obs, memory, instr_embedding=None)

[view_source]

Copied (with minor modifications) from babyai.model.ACModel.forward(...).