Skip to content

Commit 277b682

Browse files
committed
Add setup.py
1 parent 5737c20 commit 277b682

File tree

3 files changed

+6
-2
lines changed

3 files changed

+6
-2
lines changed

nn_generalisation/landscape_analysis/__init__.py

Whitespace-only changes.

nn_generalisation/landscape_analysis/slope.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,6 @@
44

55
import torch
66
import torch.nn.functional as F
7-
import matplotlib.pyplot as plt
8-
import numpy as np
97

108
def get_jacobian_avg(model : DenseNN,
119
exp : Experiment) -> float:

setup.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
from setuptools import setup
2+
3+
setup(name='nn_generalisation',
4+
version='0.1',
5+
author='David Batchelder',
6+
packages=['nn_generalisation'])

0 commit comments

Comments
 (0)