allenact_plugins.babyai_plugin.babyai_models#
BabyAIACModelWrapped#
class BabyAIACModelWrapped(babyai.model.ACModel)
BabyAIACModelWrapped.forward_once#
| forward_once(obs, memory, instr_embedding=None)
Copied (with minor modifications) from
babyai.model.ACModel.forward(...).