Skip to content

Instantly share code, notes, and snippets.

Avatar

Chirag Raman chiragraman

View GitHub Profile
@chiragraman
chiragraman / gist:16b1a89787df0c517b8dfffae5c3d591
Created Oct 2, 2020
Minimal Example for bug report for Pytorch-Lightning
View gist:16b1a89787df0c517b8dfffae5c3d591
import argparse
from pathlib import Path
from typing import Tuple
import torch
import torch.nn.functional as F
from torch import Tensor
from torch.utils.data import Dataset, DataLoader
import pytorch_lightning as pl
from pytorch_lightning import Trainer, seed_everything