Fix Bug in evaluate script
@@ -143,7 +143,12 @@ def evaluate_from_args(args: argparse.Namespace) -> Dict[str, Any]:
logging.getLogger("allennlp.modules.token_embedders.embedding").setLevel(logging.INFO)
# Load from archive
- archive = load_archive(args.archive_file, args.cuda_device, args.overrides, args.weights_file)
+ archive = load_archive(
+ args.archive_file,
+ weights_file=args.weights_file,
+ cuda_device=args.cuda_device,
+ overrides=args.overrides,
+ )
config = archive.config
prepare_environment(config)
model = archive.model
