File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -101,10 +101,10 @@ nvinfer1::ITensor* Var::ITensorOrFreeze(ConversionCtx* ctx) {
101
101
TRTORCH_THROW_ERROR (" Unable to freeze tensor of type Int64/Float64 into constant layer, try to compile model with truncate_long_and_double ON" );
102
102
} else if (tensor.scalar_type () == at::kLong && ctx->settings .truncate_long_and_double ) {
103
103
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." );
105
105
} else if (tensor.scalar_type () == at::kDouble && ctx->settings .truncate_long_and_double ) {
106
106
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." );
108
108
} else {
109
109
weights = converters::Weights (ctx, tensor);
110
110
}
You can’t perform that action at this time.
0 commit comments