Skip to content

methods decorated with compute and launched in separate process can't be interrupted #1

@krafczyk

Description

@krafczyk

Encountered when running dryml code in a jupyter notebook.

I decorated a training method with @dryml.compute and realized I had included debug print statements in a forward method of a pytorch model (which is called very frequently) and the cell output was being flooded with messages. I interrupted the cell, but realized that the subprocess doesn't receive the interrupt signal, and in fact won't stop because it waits for the closing signal from one of the queues. I had to terminate the process.

Could be more reason to go to ray for this type of execution?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions