diff --git a/model.cpp b/model.cpp index de58e0955e90e75882b16e099d8bb056a984ee2e..ce60208236846602a82ed8065d7f662d31ef790d 100644 --- a/model.cpp +++ b/model.cpp @@ -620,7 +620,7 @@ std::string Model::getTorchScript() out<<" assert parameters.size(0) is "<<parameters.size()<<"\n\n"; for(size_t i = 0; i < parameters.size(); ++i) out<<" "<<parameters[i]<<" = parameters["<<i<<"]\n"; - out<<"\n return "<<formular<<'\n'; + out<<"\n return "<<formular<<"+0*omegas\n"; return out.str(); } diff --git a/resistor.cpp b/resistor.cpp index 2318555e32bd1ebf5dfe40fe27a2e1a62ab3bb49..96817141bd09f220afb491e46579176dbe193678 100644 --- a/resistor.cpp +++ b/resistor.cpp @@ -58,5 +58,5 @@ std::string Resistor::getTorchScript(std::vector<std::string>& parameters) { parameters.push_back(getUniqueName() + "_0"); - return parameters.back(); + return parameters.back() + "*(1+0j)"; }