Fix Errors & Parameterize File Paths

- Update gitignore to exclude vscode files
- Parameterize hardcoded file paths
- Comment out references to pgf image and latex dependency
- Allow pickle in numpy.load
- Add TODO comments for where settings need to be changed
- Explicitly set visualization to False
- Change pytorch device from cuda to cpu
This commit is contained in:
Sravan Balaji
2020-04-30 19:58:05 -04:00
parent d3d6d1cd82
commit 93ac62b154
4 changed files with 65 additions and 52 deletions

View File

@@ -7,7 +7,7 @@ import skimage.measure
import torch
device = torch.device('cuda')
device = torch.device('cpu')
polesides = range(1, 5+1)
minscore = 0.6
minheight = 1.0