Skip to content

Instantly share code, notes, and snippets.

View Cheneng's full-sized avatar
🎯
Focusing

ColeChen Cheneng

🎯
Focusing
  • China
View GitHub Profile
@Cheneng
Cheneng / weight_init.py
Created July 16, 2019 06:41 — forked from jeasinema/weight_init.py
A simple script for parameter initialization for PyTorch
#!/usr/bin/env python
# -*- coding:UTF-8 -*-
import torch
import torch.nn as nn
import torch.nn.init as init
def weight_init(m):
'''