Skip to content

Instantly share code, notes, and snippets.

View janezzzz's full-sized avatar

Hang Zhang janezzzz

  • McGill University
  • Montreal
View GitHub Profile
@iamNCJ
iamNCJ / glue.py
Created March 24, 2021 07:40
GLUE with transformer and pytorch lightning
from argparse import ArgumentParser
from datetime import datetime
from typing import Optional
import datasets
import numpy as np
import pytorch_lightning as pl
import torch
from torch.utils.data import DataLoader
from transformers import (