Update predict.py

predict.py

@@ -13,13 +13,13 @@ def main(args):
     
     Net(model_arch=model_arch)
     DeepStruc = Net.load_from_checkpoint(model_path,model_arch=model_arch)
-    start_time = time.time()
+    #start_time = time.time()
     xyz_pred, latent_space, kl, mu, sigma = DeepStruc(data, mode='prior', sigma_scale=args.sigma)
-    st.write("one prediction: " , time.time() - start_time)
-    start_time = time.time()
-    for i in range(1000):
-        xyz_pred, latent_space, kl, mu, sigma = DeepStruc(data, mode='prior', sigma_scale=args.sigma)
-    st.write("thousand predictions: " , time.time() - start_time)
+    #st.write("one prediction: " , time.time() - start_time)
+    #start_time = time.time()
+    #for i in range(1000):
+    #    xyz_pred, latent_space, kl, mu, sigma = DeepStruc(data, mode='prior', sigma_scale=args.sigma)
+    #st.write("thousand predictions: " , time.time() - start_time)
 
     samling_pairs = format_predictions(latent_space, data_name, mu, sigma, args.sigma)