TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
polygraphy.tools.args.data_loader.DataLoaderArgs Class Reference
Inheritance diagram for polygraphy.tools.args.data_loader.DataLoaderArgs:
Collaboration diagram for polygraphy.tools.args.data_loader.DataLoaderArgs:

Public Member Functions

def __init__ (self, read=True)
 
def add_to_parser (self, parser)
 
def register (self, maker)
 
def check_registered (self)
 
def parse (self, args)
 
def add_to_script (self, script)
 
def get_data_loader (self)
 

Public Attributes

 model_args
 
 seed
 
 int_range
 
 float_range
 
 iterations
 
 load_inputs
 

Private Attributes

 _read
 

Constructor & Destructor Documentation

◆ __init__()

def polygraphy.tools.args.data_loader.DataLoaderArgs.__init__ (   self,
  read = True 
)

Member Function Documentation

◆ add_to_parser()

def polygraphy.tools.args.data_loader.DataLoaderArgs.add_to_parser (   self,
  parser 
)
Add arguments to a command-line parser.

Args:
    parser (argparse.ArgumentParser): The argument parser.

Reimplemented from polygraphy.tools.args.base.BaseArgs.

◆ register()

def polygraphy.tools.args.data_loader.DataLoaderArgs.register (   self,
  maker 
)
Registers another argument group with this one.
This can be used to pick up dependencies for example.

Args:
    maker (BaseArgs): Another argument group.

Reimplemented from polygraphy.tools.args.base.BaseArgs.

◆ check_registered()

def polygraphy.tools.args.data_loader.DataLoaderArgs.check_registered (   self)
Called after all `register()` calls to make dependency checks easier.

Reimplemented from polygraphy.tools.args.base.BaseArgs.

◆ parse()

def polygraphy.tools.args.data_loader.DataLoaderArgs.parse (   self,
  args 
)
Parses relevant arguments from command-line arguments.

Args:
    args: Arguments provided by argparse.

Reimplemented from polygraphy.tools.args.base.BaseArgs.

◆ add_to_script()

def polygraphy.tools.args.data_loader.DataLoaderArgs.add_to_script (   self,
  script 
)
Here is the caller graph for this function:

◆ get_data_loader()

def polygraphy.tools.args.data_loader.DataLoaderArgs.get_data_loader (   self)
Here is the call graph for this function:

Member Data Documentation

◆ _read

polygraphy.tools.args.data_loader.DataLoaderArgs._read
private

◆ model_args

polygraphy.tools.args.data_loader.DataLoaderArgs.model_args

◆ seed

polygraphy.tools.args.data_loader.DataLoaderArgs.seed

◆ int_range

polygraphy.tools.args.data_loader.DataLoaderArgs.int_range

◆ float_range

polygraphy.tools.args.data_loader.DataLoaderArgs.float_range

◆ iterations

polygraphy.tools.args.data_loader.DataLoaderArgs.iterations

◆ load_inputs

polygraphy.tools.args.data_loader.DataLoaderArgs.load_inputs

The documentation for this class was generated from the following file: