Add demo file

jitdee-ai · Oct 27, 2020 · 3cf04b2 · 3cf04b2
1 parent 5027eed
commit 3cf04b2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/test.py → demo.py b/test.py → demo.py
@@ -1,7 +1,5 @@
 import darmo
-import torch
-from collections import OrderedDict
 
 model = darmo.create_model("relative_nas", num_classes=1000, pretrained=True)
 model.reset_classifier(num_classes=100, dropout=0.2)
-print(model)
+print(model)