From a56f93d2b5f52f6c576177a94f428d4ec507bab3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E6=99=A8?= Date: Thu, 17 Aug 2023 11:21:39 +0800 Subject: [PATCH] udpate --- ieemoo-ai-conpurchase.py | 2 +- network/BaseNet.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/ieemoo-ai-conpurchase.py b/ieemoo-ai-conpurchase.py index dd24613..0f64cd2 100644 --- a/ieemoo-ai-conpurchase.py +++ b/ieemoo-ai-conpurchase.py @@ -10,7 +10,7 @@ from utils.embedding import DataProcessing as dp app = Flask(__name__) parser = argparse.ArgumentParser() -parser.add_argument('--model', default='./checkpoint/raft-small.pth',help="restore checkpoint") +parser.add_argument('--model', default='../module/ieemoo-ai-conpurchase/model/now/raft-small.pth',help="restore checkpoint") parser.add_argument('--checkpoint', default='mobilevit',help="get embedding ") parser.add_argument('--device', default='cuda',help="device") parser.add_argument('--small', type=bool, default=True, help='use small model') diff --git a/network/BaseNet.py b/network/BaseNet.py index 56484b9..d4406c7 100644 --- a/network/BaseNet.py +++ b/network/BaseNet.py @@ -170,7 +170,6 @@ if __name__ == '__main__': img = transform(img) img = img.cuda() - # from torchsummary import summary device = torch.device("cuda" if torch.cuda.is_available() else "cpu") model = ResnetFpn().to(device)