Hugging Face's logo

Spaces:
SuyeonJ
/
VfiTest
Sleeping

App Files Community
SuyeonJ commited on
Commit
f7a7cfe
·
verified ·
1 Parent(s): 3927cee

Upload folder using huggingface_hub

Browse files
Files changed (11) hide show
  1. main.py +1 -1
  2. modules/components/m2m_flow_former/costvol.py +1 -1
  3. modules/components/m2m_flow_former/softsplat.py +1 -1
  4. modules/components/m2m_pwc/costvol.py +1 -1
  5. modules/components/m2m_pwc/softsplat.py +1 -1
  6. modules/components/m2m_unimatch/costvol.py +1 -1
  7. modules/components/m2m_unimatch/softsplat.py +1 -1
  8. modules/components/upr_net_freq/costvol.py +1 -1
  9. modules/components/upr_net_mod/costvol.py +1 -1
  10. modules/components/upr_net_mod2/costvol.py +1 -1
  11. modules/components/upr_net_multi_flow/softsplat.py +1 -1
main.py CHANGED
@@ -47,7 +47,7 @@ def make_cfg(args):
47
  env = dict()
48
  env['exp_name'] = exp_name + '_' + cfg['exp_name']
49
  env['save_dir'] = os.path.join(args.save_root, env['exp_name'])
50
- env['tot_gpus'] = torch.cuda.device_count()
51
  env['cudnn'] = args.cudnn
52
  env['port'] = str(29600 + args.port_offset)
53
  env['wandb_upload'] = args.wandb_upload
 
47
  env = dict()
48
  env['exp_name'] = exp_name + '_' + cfg['exp_name']
49
  env['save_dir'] = os.path.join(args.save_root, env['exp_name'])
50
+ #env['tot_gpus'] = torch.cuda.device_count()
51
  env['cudnn'] = args.cudnn
52
  env['port'] = str(29600 + args.port_offset)
53
  env['wandb_upload'] = args.wandb_upload
modules/components/m2m_flow_former/costvol.py CHANGED
@@ -26,7 +26,7 @@ def cuda_float32(fltIn:float):
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
- objCudacache['device'] = torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
 
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
modules/components/m2m_flow_former/softsplat.py CHANGED
@@ -31,7 +31,7 @@ def cuda_float32(fltIn:float):
31
 
32
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
33
  if 'device' not in objCudacache:
34
- objCudacache['device'] = torch.cuda.get_device_name()
35
  # end
36
 
37
  strKey = strFunction
 
31
 
32
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
33
  if 'device' not in objCudacache:
34
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
35
  # end
36
 
37
  strKey = strFunction
modules/components/m2m_pwc/costvol.py CHANGED
@@ -26,7 +26,7 @@ def cuda_float32(fltIn:float):
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
- objCudacache['device'] = torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
 
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
modules/components/m2m_pwc/softsplat.py CHANGED
@@ -31,7 +31,7 @@ def cuda_float32(fltIn:float):
31
 
32
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
33
  if 'device' not in objCudacache:
34
- objCudacache['device'] = torch.cuda.get_device_name()
35
  # end
36
 
37
  strKey = strFunction
 
31
 
32
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
33
  if 'device' not in objCudacache:
34
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
35
  # end
36
 
37
  strKey = strFunction
modules/components/m2m_unimatch/costvol.py CHANGED
@@ -26,7 +26,7 @@ def cuda_float32(fltIn:float):
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
- objCudacache['device'] = torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
 
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
modules/components/m2m_unimatch/softsplat.py CHANGED
@@ -31,7 +31,7 @@ def cuda_float32(fltIn:float):
31
 
32
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
33
  if 'device' not in objCudacache:
34
- objCudacache['device'] = torch.cuda.get_device_name()
35
  # end
36
 
37
  strKey = strFunction
 
31
 
32
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
33
  if 'device' not in objCudacache:
34
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
35
  # end
36
 
37
  strKey = strFunction
modules/components/upr_net_freq/costvol.py CHANGED
@@ -26,7 +26,7 @@ def cuda_float32(fltIn:float):
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
- objCudacache['device'] = torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
 
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
modules/components/upr_net_mod/costvol.py CHANGED
@@ -26,7 +26,7 @@ def cuda_float32(fltIn:float):
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
- objCudacache['device'] = torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
 
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
modules/components/upr_net_mod2/costvol.py CHANGED
@@ -26,7 +26,7 @@ def cuda_float32(fltIn:float):
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
- objCudacache['device'] = torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
 
26
 
27
  def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
28
  if 'device' not in objCudacache:
29
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
30
  # end
31
 
32
  strKey = strFunction
modules/components/upr_net_multi_flow/softsplat.py CHANGED
@@ -34,7 +34,7 @@ def cuda_float32(fltIn: float):
34
 
35
  def cuda_kernel(strFunction: str, strKernel: str, objVariables: typing.Dict):
36
  if 'device' not in objCudacache:
37
- objCudacache['device'] = torch.cuda.get_device_name()
38
  # end
39
 
40
  strKey = strFunction
 
34
 
35
  def cuda_kernel(strFunction: str, strKernel: str, objVariables: typing.Dict):
36
  if 'device' not in objCudacache:
37
+ objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
38
  # end
39
 
40
  strKey = strFunction