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) {
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 }
You can’t perform that action at this time.
0 commit comments