From 3cf04b27c14828c3b6a8fc1b5aabd4bd7a18f1bd Mon Sep 17 00:00:00 2001 From: Chakkrit Termritthikun Date: Wed, 28 Oct 2020 01:03:25 +1030 Subject: [PATCH] Add demo file Add demo file --- test.py => demo.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) rename test.py => demo.py (70%) diff --git a/test.py b/demo.py similarity index 70% rename from test.py rename to demo.py index 24c2850..b9f63de 100644 --- a/test.py +++ b/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) \ No newline at end of file