Skip to content
This repository was archived by the owner on Nov 28, 2023. It is now read-only.
This repository was archived by the owner on Nov 28, 2023. It is now read-only.

[potential bug]  #210

Description

@LilySnow

From @manonreau on graphprot:

When you store loss values, do not forget to detach it from the computation graph (that is used for the back propagation) or you could run into memory issues ! :)
ex :

tot_loss += loss.item()          # DON'T DO THAT
tot_loss += loss.detach().item() # DO THAT :D

DeepRank is using:
running_loss += loss.data.item() in NeuralNet.py. Maybe we want to correct it.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions