update ranker_test.pth

app.py

@@ -9,7 +9,7 @@ from Model.fur_rl.models.retriever_rl import DQN_v3
 import Model.CLIP.cn_clip.clip as clip
 import recommendation.datasets.img_preprocess
 import recommendation.utils.ranker_1
-import numpy as np
+
 
 st.title('Recommendation System V1')
 
@@ -63,6 +63,7 @@ def start(modelName):
                 load_model_name = None
             net.actor_net.load_state_dict(torch.load(load_model_name, map_location=device1)['actor_state_dict'])
             net.actor_optimizer.load_state_dict(torch.load(load_model_name, map_location=device1)['actor_optimizer'])
+            net.actor_net.eval()
             st.session_state['model'] = modelName
             st.session_state['net'] = net
             st.write(modelName + ' reloaded')
@@ -92,12 +93,16 @@ def start(modelName):
     if 'ranker' not in st.session_state:
         test_img_ids = "./recommendation/datasets/test_img_id_r.csv"
         dataset_test = recommendation.datasets.img_preprocess.Image_preprocess(test_img_ids)
-        dataloader_test = torch.utils.data.DataLoader(dataset_test, batch_size=1, shuffle=False, num_workers=4)
         ranker_test = recommendation.utils.ranker_1.Ranker(device1, dataset_test, batch_size=64)
-        st.session_state['ranker'] = ranker_test
-        net = st.session_state['net']
-        ranker_test.update_emb(model=net.actor_net)  # 220.0789999961853s; 78s on 3090
+        # net = st.session_state['net']
+        # ranker_test.update_emb(model=net.actor_net)  # 220.0789999961853s; 78s on 3090
+        # save ranker
+        # torch.save(ranker_test, './ranker_test.pth')
+        # load ranker from pth
+        ranker_test = torch.load('./models/ranker_test.pth', map_location=device1)
         st.write('ranker ready')
+        st.session_state['ranker'] = ranker_test
+
 
     # st.write(st.session_state)
     st.write(st.session_state['model'] + ' loaded')

models/ranker_test.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8c8c80fce74913e38a3b0c20e7ff06db179ba069c945c504c18af2fba445ba9a
+oid sha256:53a0ab3288b8fda1c8a9599b50b9bccf0a8d33ab309a5ef3742e49ec419c6ffb
 size 1703781047