Skip to content

Commit e7608c8

Browse files
committed
update warning infomation
Signed-off-by: inocsin <[email protected]>
1 parent 60dba12 commit e7608c8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

core/conversion/var/Var.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,10 +101,10 @@ nvinfer1::ITensor* Var::ITensorOrFreeze(ConversionCtx* ctx) {
101101
TRTORCH_THROW_ERROR("Unable to freeze tensor of type Int64/Float64 into constant layer, try to compile model with truncate_long_and_double ON");
102102
} else if (tensor.scalar_type() == at::kLong && ctx->settings.truncate_long_and_double) {
103103
weights = converters::Weights(ctx, tensor.toType(at::kInt));
104-
LOG_WARNING("Warning: Truncating weight (constant in the graph) from Int64 to Int32.");
104+
LOG_WARNING("Truncating weight (constant in the graph) from Int64 to Int32.");
105105
} else if (tensor.scalar_type() == at::kDouble && ctx->settings.truncate_long_and_double) {
106106
weights = converters::Weights(ctx, tensor.toType(at::kFloat));
107-
LOG_WARNING("Warning: Truncating weight (constant in the graph) from Float64 to Float32.");
107+
LOG_WARNING("Truncating weight (constant in the graph) from Float64 to Float32.");
108108
} else {
109109
weights = converters::Weights(ctx, tensor);
110110
}

0 commit comments

Comments
 (0)