torch._dynamo.exc.UncapturedHigherOrderOpError: Cond doesn't work unless it is captured completely with torch.compile. Scroll up to find out what causes the graph break. It works without print. Then I did (look for added line): import torch class SubThen(torch.nn.Module): def forward(...
compile_et complete composite conjure conman conmand continue convert convertquota cp cpan cpio cpp cpuspeed crash createdb createlang createuser cron crond crontab cryptsetup csh csplit ctags cupsctl cupsd curl cut cvs cvsbug cyradm cytune D date dc dd deallocvt debugfs declare defer deliver delpart...
fromtorch.utils._tracebackimportreport_compile_source_on_error fromtorch.utils._tracebackimportreport_compile_source_on_error,format_traceback_short importtorch.utils.cpp_extension fromtorch.autograd._functions.utilsimportcheck_onnx_broadcast fromtorch.onnx.symbolic_opset9import_prepare_onnx_paddings @@...