Views
No views yet
1from train import ActorCritic 2import torch 3 4model = ActorCritic(obs_dim=96, max_neighbors=16, hidden_sizes=[512, 512, 256]) 5model.load_state_dict(torch.load("best_model.pt", weights_only=True)) 6model.eval()