JianyuanWang commited on
Commit
7d141d7
1 Parent(s): c2c23af
Files changed (2) hide show
  1. app.py +1 -0
  2. vggsfm_code/cfgs/demo.yaml +1 -1
app.py CHANGED
@@ -30,6 +30,7 @@ def vggsfm_demo(
30
  query_frame_num,
31
  max_query_pts=4096,
32
  ):
 
33
  torch.cuda.empty_cache()
34
 
35
  if input_video is not None:
 
30
  query_frame_num,
31
  max_query_pts=4096,
32
  ):
33
+ gc.collect()
34
  torch.cuda.empty_cache()
35
 
36
  if input_video is not None:
vggsfm_code/cfgs/demo.yaml CHANGED
@@ -10,7 +10,7 @@ viz_ip: 127.0.0.1
10
  debug: False
11
 
12
  center_order: True
13
- mixed_precision: fp16
14
  extract_color: True
15
  filter_invalid_frame: True
16
 
 
10
  debug: False
11
 
12
  center_order: True
13
+ mixed_precision: bf16
14
  extract_color: True
15
  filter_invalid_frame: True
16