Skip to content

Instantly share code, notes, and snippets.

@Ushk
Ushk / gist:96be97cd595ee63ae5cd6cebe415e642
Last active October 15, 2020 22:07
Pytorch1.5_Cuda10.1_timing.py
import random
import os
import numpy as np
import time
import torch
import torch.nn as nn
from apex import amp
def set_seed(seed: int):
"""Set all seeds to make results reproducible (deterministic mode).
<!--You must include this JavaScript file -->
<script src="https://assets.crowd.aws/crowd-html-elements.js"></script>
<script type='text/javascript'>
document.getElementById('theImage').style.visibility='hidden'
</script>
<script type='text/javascript'>
function setImageVisible(id, visible) {
@Ushk
Ushk / dp_segfault.py
Last active September 15, 2018 14:49
Demo Script to show Segmentation Fault in Pytorch DataParallel & Checkpoint
import torch
import torch.nn as nn
from torch import optim
import torch.utils.checkpoint as chk
import torch.nn.functional as F
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0,3'
import faulthandler