Skip to content

Instantly share code, notes, and snippets.

View ruotianluo's full-sized avatar

Ruotian(RT) Luo ruotianluo

View GitHub Profile
@ruotianluo
ruotianluo / rl-tutorial-1.ipynb
Created June 16, 2016 04:19 — forked from awjuliani/rl-tutorial-1.ipynb
Reinforcement Learning Tutorial 1 (Two-armed bandit problem)
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
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
ruotianluo / test_roialign.py
Created October 6, 2017 02:22
A snippet to show how roialign works
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
ruotianluo / Detectron.pytorch_pth1_patch.diff
Created February 10, 2019 04:01
Pytorch 1.0 patch for Detectron.pytorch
This file has been truncated, but you can view the full file.
From 784c05a1ca663d88d68a644108ad2514b79310dc Mon Sep 17 00:00:00 2001
From: adityaarun1 <adityaarun1@gmail.com>
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 +
# This file contains Transformer network
# Most of the code is copied from http://nlp.seas.harvard.edu/2018/04/03/attention.html
# The cfg name correspondance:
# N=num_layers
# d_model=input_encoding_size
# d_ff=rnn_size
# h is always 8
from __future__ import absolute_import
@ruotianluo
ruotianluo / train_cityscapes2.py
Created May 19, 2020 21:30
train_cityscapes2.py
import torch
import torch.multiprocessing as mp
import torch.distributed as dist
from torchvision import transforms
import random
from functools import partial
from easydict import EasyDict as edict
from albumentations import (
Compose, HorizontalFlip, ShiftScaleRotate, PadIfNeeded, RandomCrop,
RGBShift, RandomBrightness, RandomContrast
@ruotianluo
ruotianluo / train_cityscapes2.py
Created May 19, 2020 21:30
train_cityscapes2.py
import torch
import torch.multiprocessing as mp
import torch.distributed as dist
from torchvision import transforms
import random
from functools import partial
from easydict import EasyDict as edict
from albumentations import (
Compose, HorizontalFlip, ShiftScaleRotate, PadIfNeeded, RandomCrop,
RGBShift, RandomBrightness, RandomContrast
@ruotianluo
ruotianluo / train_cityscapes2.py
Created May 19, 2020 21:30
train_cityscapes2.py
import torch
import torch.multiprocessing as mp
import torch.distributed as dist
from torchvision import transforms
import random
from functools import partial
from easydict import EasyDict as edict
from albumentations import (
Compose, HorizontalFlip, ShiftScaleRotate, PadIfNeeded, RandomCrop,
RGBShift, RandomBrightness, RandomContrast
import os
import tempfile
import torch
import torch.distributed as dist
import torch.nn as nn
import torch.optim as optim
import torch.multiprocessing as mp
from torch.nn.parallel import DistributedDataParallel as DDP
import torch
import torch.nn as nn
class X(nn.Module):
def __init__(self):
super().__init__()
self.a = nn.Linear(3,4)
def forward(self, x):
x = next(self.parameters())
import torchvision