Commit 554b800e authored by Devanshu Agrawal's avatar Devanshu Agrawal
Browse files

Fix tempmodules to modules.

parent 16c4e17b

torch_experiments/main.py

+1 −1
Original line number Diff line number Diff line
@@ -5,7 +5,7 @@ import argparse 
import torch

import torch.optim as optim

from torch.utils.data import DataLoader

from tempmodules import BottleneckNNGP

from modules import BottleneckNNGP

import datasets



# set torch seed