4898e32fbb0c5f3414c8e05c392c4cb7360b0b8f,test/test_pytorchexperimentlogger.py,,,#,250

Before Change

        return F.log_softmax(x, dim=1)


if __name__ == "__main__":
    unittest.main()

After Change

import shutil
import time
import matplotlib
matplotlib.use("Agg")

import numpy as np

import torch

Instances