Views
No views yet
1import torch
2# Assumes you have the ActorCritic class defined in your script
3config = Config() # Using your existing Config class
4model = ActorCritic(input_channels=4, action_dim=6) # Space Invaders has 6 actions
5model.load_state_dict(torch.load('ppo_final_10M.pt', map_location='cpu'))
6model.eval()