Came across the problem, that in pytorch 1.1.0 with uwsgi+flask on cpu even
torch.cat does not work (everything just freezes without errors). I determined that problem is in uwsgi/flask, because in the same environment I was able to do the same operations without any issues. There was no such problem in previous versions of pytorch. As far as I understand, I am dealing with multiprocessing artifacts…
Nevertheless I found the solution and it was simple:
app = flask.Flask(__name__) segmentator = None @app.before_first_request def load_segmentator(): global segmentator segmentator = Segmentator()
Segmentator is a class with pytorch’s
nn.Module, which loads weights in
Hope, it will help somebody.
P.S. If someone explains what is going on here, will be grateful.