Closed hai-h-nguyen closed 2 years ago
Hi @hai-h-nguyen,
This definitely should be possible depending on the type of visualization you're looking for. A few quick questions:
Some initial pointers:
InferenceAgent
class to run your model on a task in python while avoiding all the AllenAct training/evaluation infrastructure. Note that you'd need to pass the path to your model checkpoint in the call to InferenceAgent.from_experiment_config(...)
as currently it just runs with a randomly initialized agent. Once you have this running you can visualize observations
returned from the task however you like.Hi, thanks for a quick reply.
Thanks, I am able to visualize the policy using InferenceAgent.
Hello,
Is it supported to visualize a trained policy from a checkpoint? If yes, how would I do that? I saw there was a tutorial on how to evaluate a checkpoint, but I want to visually inspect the learned policy too.