Skip to content
Permalink
Browse files

Removes ununsed multi-class PSPNet model

  • Loading branch information...
daniel-j-h committed Oct 24, 2018
1 parent 8e99e5b commit a8e0e3d676b454b61df03897e43e003867b6ef48
Showing with 0 additions and 152 deletions.
  1. +0 −149 robosat/pspnet.py
  2. +0 −3 robosat/tools/train.py

This file was deleted.

Oops, something went wrong.
@@ -62,9 +62,6 @@ def main(args):
if model["common"]["cuda"] and not torch.cuda.is_available():
sys.exit("Error: CUDA requested but not available")

# if args.batch_size < 2:
# sys.exit('Error: PSPNet requires more than one image for BatchNorm in Pyramid Pooling')

os.makedirs(model["common"]["checkpoint"], exist_ok=True)

num_classes = len(dataset["common"]["classes"])

0 comments on commit a8e0e3d

Please sign in to comment.
You can’t perform that action at this time.