Skip to content

Instantly share code, notes, and snippets.

Avatar
🎯
Focusing

陈诚 Cheneng

🎯
Focusing
View GitHub Profile
@Cheneng
Cheneng / weight_init.py
Created Jul 16, 2019 — forked from jeasinema/weight_init.py
A simple script for parameter initialization for PyTorch
View weight_init.py
#!/usr/bin/env python
# -*- coding:UTF-8 -*-
import torch
import torch.nn as nn
import torch.nn.init as init
def weight_init(m):
'''
You can’t perform that action at this time.