ruotianluo

From 784c05a1ca663d88d68a644108ad2514b79310dc Mon Sep 17 00:00:00 2001
From: adityaarun1 <[email protected]>
Date: Wed, 16 Jan 2019 22:15:26 +0530
Subject: [PATCH] migrating to Pytorch-1.0

---
 .gitignore                                    |    2 +
 README.md                                     |   13 +-
 lib/make.sh                                   |   56 +-
 lib/model/csrc/ROIAlign.h                     |   46 +

ruotianluo

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable

import numpy as np

ruotianluo

import torch
import time

def timeit():
  torch.cuda.synchronize()
  start = time.time()
  x = torch.cuda.FloatTensor(10000,10000)
  torch.cuda.synchronize()
  print(time.time() - start)
  return x

ruotianluo