Engine.fit can't trained in .py file, but can run in .ipynb file

[kekaipr]

Hi, I'm curious about why this problem will happend
The following code can run in my .ipynb file, but when I put it in another .py file, it never stop and also didn't show any response

engine.fit(datamodule=folder_datamodule, model=model)

Also, I have some problem when I'm using Cflow model, the error is as follow.

RuntimeError: Exception from src\inference\src\cpp\core.cpp:109: Exception from src\inference\src\dev\plugin.cpp:54: Check 'dim.compatible(1)' failed at src\core\shape_inference\include\squeeze_shape_inference.hpp:104: While validating node 'opset1::Squeeze Squeeze_127385 (opset1::Maximum Maximum_127383[0]:f32[64], opset1::Constant Constant_127384[0]:i64[]) -> (dynamic[...])' with friendly_name 'Squeeze_127385': provided axis value is invalid. Only axes of size 1 may be removed.

If I did something wrong, please tell me, thank you for your kindness.