fix: imports in __init__

archinetai · Apr 19, 2022 · 5391597 · 5391597
1 parent 75ea27a
commit 5391597
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -5,6 +5,15 @@
 
 A PyTorch implementation of <a href="https://aclanthology.org/2020.acl-main.197.pdf">SMART</a>, a regularization technique to better fine-tune pretrained language models. 
 
+## Install
+
+```bash
+$ pip install smart-pytorch
+```
+
+## Usage
+
+
 ## Citations
 
 ```bibtex

diff --git a/smart_pytorch/__init__.py b/smart_pytorch/__init__.py
@@ -0,0 +1 @@
+from smart_pytorch.smart_pytorch import SMARTLoss
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from smart_pytorch.smart_pytorch import SMARTLoss