udpate

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')

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)