Skip to content

Instantly share code, notes, and snippets.

View dongkwan-kim's full-sized avatar
🎯
Focusing

Dongkwan Kim dongkwan-kim

🎯
Focusing
View GitHub Profile
import random
T = 596
R, C = 37, 16
numbers = list(range(1, T + 1))
random.shuffle(numbers)
sets = []
for _set in range(R):
from typing import Tuple, List
try:
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.ticker import LinearLocator
from mpl_toolkits.mplot3d import Axes3D
except ImportError:
pass
import numpy as np
@dongkwan-kim
dongkwan-kim / latex_patterns.tex
Created October 4, 2021 08:51
My LaTeX patterns
\newcommand\todo[1]{\textcolor{red}{#1}}
\newcommand\g[1]{\textcolor{gray}{#1}}
\newcommand*{\scale}[2][4]{\scalebox{#1}{$#2$}}%
\newcommand*{\resize}[2]{\resizebox{#1}{!}{$#2$}}%
\newcommand\ko[1]{
\begin{CJK}{UTF8}{} % require \usepackage{CJKutf8}
\CJKfamily{mj}
#1
\end{CJK}
}
from typing import Dict, Any, List, Tuple
import torch
from torch import Tensor
def auto_index_select(value_tensor: Tensor, index_tensor: Tensor):
index_tensor = index_tensor.squeeze()
sizes = value_tensor.size()
for dim, dim_size in enumerate(sizes):
%%%%% NEW MATH DEFINITIONS %%%%%
\usepackage{amsmath,amsfonts,bm}
% Mark sections of captions for referring to divisions of figures
\newcommand{\figleft}{{\em (Left)}}
\newcommand{\figcenter}{{\em (Center)}}
\newcommand{\figright}{{\em (Right)}}
\newcommand{\figtop}{{\em (Top)}}
\newcommand{\figbottom}{{\em (Bottom)}}
nvidia-smi --query-compute-apps=pid,process_name,used_memory --format=csv
ps all --sort=-time | grep python
python -u analysis.py | tee log.txt
ps aux | grep -ie "query" | awk '{print $2}' | xargs kill -9
# https://github.com/golbin/TensorFlow-Tutorials/blob/master/06%20-%20MNIST/01%20-%20MNIST.py
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("./mnist/data/", one_hot=True)
X = tf.placeholder(tf.float32, [None, 784])
Y = tf.placeholder(tf.float32, [None, 10])
from utils import *
params = {}
def create_variable(scope, name, shape, trainable=True, on_cpu=True, **kwargs) -> tf.Variable:
def _create_variable():
with tf.variable_scope(scope):
_w = tf.get_variable(name, shape, trainable=trainable, **kwargs)
params[_w.name] = _w
@dongkwan-kim
dongkwan-kim / nvidia_docker_setup.sh
Last active February 20, 2024 07:51
nvidia docker setup in remote server accessible by ssh
# In the server, (images what I mostly used)
docker pull nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.04
docker pull nvidia/cuda:11.0.3-cudnn8-devel-ubuntu18.04
docker pull nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04
# cuda 10.0 multiple ports volume binding
docker run -ti --runtime=nvidia --name dongkwan -p 8082:22 -p 8083:6006 --ipc=host -v /mnt/nas2:/mnt/nas2 -d nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04 /bin/bash
# Unknown runtime specified nvidia
docker run -ti --gpus '"device=0,1,2,3"' --name dongkwan -p 8070:22 --ipc=host -d nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 /bin/bash