From b95e141fbcddf601e5f872fb0712a2fa25336cf0 Mon Sep 17 00:00:00 2001 From: izhigal Date: Fri, 26 Jan 2024 15:52:42 +0800 Subject: [PATCH] more formatting --- examples/evaluate.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/examples/evaluate.py b/examples/evaluate.py index 575aa5c1..d79a54b1 100644 --- a/examples/evaluate.py +++ b/examples/evaluate.py @@ -1,19 +1,16 @@ -"""An example of evluating the trained models in RLCard -""" +"""An example of evaluating the trained models in RLCard""" import os import argparse import rlcard -from rlcard.agents import ( - DQNAgent, - RandomAgent, -) + from rlcard.utils import ( get_device, set_seed, tournament, ) + def load_model(model_path, env=None, position=None, device=None): if os.path.isfile(model_path): # Torch model import torch @@ -29,14 +26,14 @@ def load_model(model_path, env=None, position=None, device=None): else: # A model in the model zoo from rlcard import models agent = models.load(model_path).agents[position] - + return agent -def evaluate(args): +def evaluate(args): # Check whether gpu is available device = get_device() - + # Seed numpy, torch, random set_seed(args.seed) @@ -54,6 +51,7 @@ def evaluate(args): for position, reward in enumerate(rewards): print(position, args.models[position], reward) + if __name__ == '__main__': parser = argparse.ArgumentParser("Evaluation example in RLCard") parser.add_argument( @@ -99,4 +97,3 @@ def evaluate(args): os.environ["CUDA_VISIBLE_DEVICES"] = args.cuda evaluate(args) -