Skip to content

Instantly share code, notes, and snippets.

View cri5Castro's full-sized avatar
🤓
studying

Cristopher Castro cri5Castro

🤓
studying
View GitHub Profile
@jeasinema
jeasinema / weight_init.py
Last active July 30, 2024 16:35
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):
'''