bartduis commited on
Commit
f5865e1
·
verified ·
1 Parent(s): ae065ad

Update eval_wrapper/eval.py

Browse files
Files changed (1) hide show
  1. eval_wrapper/eval.py +5 -5
eval_wrapper/eval.py CHANGED
@@ -16,11 +16,11 @@ import argparse
16
  from utils import misc
17
  import torch.distributed as dist
18
  from utils.collate import collate
19
- from engine import eval_model
20
- from utils.viz import just_load_viz
21
- from utils.geometry import compute_pointmap_torch
22
- from eval_wrapper.eval_utils import filter_all_masks
23
- from huggingface_hub import hf_hub_download
24
 
25
  class EvalWrapper(torch.nn.Module):
26
  def __init__(self,checkpoint_path,distributed=False,device="cpu",dtype=torch.float32,**kwargs):
 
16
  from utils import misc
17
  import torch.distributed as dist
18
  from utils.collate import collate
19
+ # from engine import eval_model
20
+ # from utils.viz import just_load_viz
21
+ # from utils.geometry import compute_pointmap_torch
22
+ # from eval_wrapper.eval_utils import filter_all_masks
23
+ # from huggingface_hub import hf_hub_download
24
 
25
  class EvalWrapper(torch.nn.Module):
26
  def __init__(self,checkpoint_path,distributed=False,device="cpu",dtype=torch.float32,**kwargs):