def get_logits(
context,
model_snapshot,
test_path,
use_cuda,
output_path,
field_names,
dump_raw_input,
batch_size,
ndigits_precision,
output_columns,
use_gzip,
device_id,
):
"""print logits from a trained model snapshot to output_path"""
model_snapshot, use_cuda, _ = _get_model_snapshot(context, model_snapshot,
use_cuda, False)
if output_columns:
output_columns = [int(x) for x in output_columns.split(",")]
print("\n=== Starting get_logits...")
workflow_get_logits(
model_snapshot,
use_cuda,
output_path,
test_path,
field_names,
dump_raw_input,
batch_size,
ndigits_precision,
output_columns,
use_gzip,
device_id,
)
def get_logits(context, model_snapshot, test_path, use_cuda, output_path, field_names):
"""print logits from a trained model snapshot to output_path
"""
model_snapshot, use_cuda, _ = _get_model_snapshot(
context, model_snapshot, use_cuda, False
)
print("\n=== Starting get_logits...")
workflow_get_logits(model_snapshot, use_cuda, output_path, test_path, field_names)