Skip to content

Instantly share code, notes, and snippets.

View leimao's full-sized avatar
🦤
Hello Underworld. Hello 人工稚能.

Lei Mao leimao

🦤
Hello Underworld. Hello 人工稚能.
View GitHub Profile
@leimao
leimao / onnx_device.py
Created November 12, 2019 04:06
Run ONNX model on different devices
import onnxruntime
import numpy as np
import time
from onnxruntime.datasets import get_example
# ONNXRuntime API Documentation
# https://microsoft.github.io/onnxruntime/python/api_summary.html
log_severity_level = 0
model_name = "logreg_iris.onnx"
@leimao
leimao / ClearInputBuff.cpp
Created December 30, 2019 07:44 — forked from manadream/ClearInputBuff.cpp
Clearing input buffer in C++ completely
#include <iostream>
#include <limits>
using namespace std;
int main(){
char var[10];
bool valid = false;
while(!valid){
cout << "Enter a string 9 characters long: ";
@leimao
leimao / class.cpp
Created January 7, 2020 17:39
Constructor with default arguments. `g++ class.cpp test.cpp -o test`
#include "class.h"
Test::Test(int a, int b): mA{a}, mB{b}
{
}
int Test::getA()
{
return this->mA;
}
@leimao
leimao / chat_client.py
Created August 30, 2020 19:57
EdgeDB AsyncIO Tutorial ChatBot: https://youtu.be/SyiTd4rLb2s
from __future__ import annotations
from typing import IO
import asyncio
import sys
import contextlib
import aiofiles.threadpool
from chat_streams import split_lines, write, handle_writes
async def handle_reads(reader: asyncio.StreamReader) -> None:
@leimao
leimao / coco.sh
Created May 5, 2021 17:25 — forked from mkocabas/coco.sh
Download COCO dataset. Run under 'datasets' directory.
mkdir coco
cd coco
mkdir images
cd images
wget http://images.cocodataset.org/zips/train2017.zip
wget http://images.cocodataset.org/zips/val2017.zip
wget http://images.cocodataset.org/zips/test2017.zip
wget http://images.cocodataset.org/zips/unlabeled2017.zip
@leimao
leimao / resnet_ddp.py
Created June 20, 2021 02:34
PyTorch Distributed Data-Parallel Training for ResNet
import torch
from torch.utils.data.distributed import DistributedSampler
from torch.utils.data import DataLoader
import torch.nn as nn
import torch.optim as optim
import torchvision
import torchvision.transforms as transforms
import argparse
@leimao
leimao / graph.py
Last active December 15, 2022 09:15
Graph Traversal
from typing import Iterable, List, Union, Set
class Graph:
def __init__(self) -> None:
# Adjacency set representation.
# key: node name, value: adjacency node name set.
self.edges = dict()
import torch
from torch.utils.data.distributed import DistributedSampler
from torch.utils.data import DataLoader
import torch.nn as nn
import torch.optim as optim
import torchvision
import torchvision.transforms as transforms
import argparse
@leimao
leimao / pytorch_mnist_2.py
Created June 30, 2021 01:34
Horovod PyTorch
import argparse
import os
from filelock import FileLock
import torch.multiprocessing as mp
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torchvision import datasets, transforms
import torch.utils.data.distributed
mkdir -p n01440764
mkdir -p n01443537
mkdir -p n01484850
mkdir -p n01491361
mkdir -p n01494475
mkdir -p n01496331
mkdir -p n01498041
mkdir -p n01514668
mkdir -p n01514859
mkdir -p n01518878