Commit ab5c627c authored by E144069X's avatar E144069X

Added try except block to prevent OSError

parent f17404f0
......@@ -17,8 +17,11 @@ import matplotlib.pyplot as plt
#But to avoid program stopping when doing test on a no-gpu machine
#the error is catched
try:
import torch_geometric
import pointnet2
try:
import torch_geometric
import pointnet2
except OSError:
pass
except ModuleNotFoundError:
pass
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment