Skip to content

Instantly share code, notes, and snippets.

View NinoRisteski's full-sized avatar

Nino Risteski NinoRisteski

View GitHub Profile
from dataclasses import dataclass
import torch
import torch.nn as nn
import torch.nn.functional as F
import math
import os
class LayerNorm(nn.Module):
""" LayerNorm but with an optional bias. PyTorch doesn't support simply bias=False """