We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 55def2e commit 87ac706Copy full SHA for 87ac706
dlclive/benchmark_pytorch.py
@@ -1,5 +1,4 @@
1
import csv
2
-import os
3
import platform
4
import subprocess
5
import sys
@@ -8,10 +7,11 @@
8
7
import colorcet as cc
9
import cv2
10
import h5py
11
-import numpy as np
12
-import torch
13
from PIL import ImageColor
14
from pip._internal.operations import freeze
+import torch
+# torch import needs to switch order with "from pip._internal.operations import freeze" because of crash
+# see https://github.com/pytorch/pytorch/issues/140914
15
16
from dlclive import DLCLive
17
from dlclive.version import VERSION
0 commit comments