Stuck at `nn.init.orthogonal_()`

In a pytorch model, one of the line is to initialise the weights of a module using orthogonal. But sometimes, on Linux, when initialising the model which internally initialises the modules, the system is stuck at the init orthogonal_ line for a very long time.

Here is the model code, which is calling another module.

The orthogonal_ is being called in this module. I had to add print statement above and below every line, to check where exactly the code is getting stuck, and at this line is where it gets stuck.

Any help would be highly appreciated.