Disable warnings

pytorch · Feb 21, 2023 · b5e3b91 · b5e3b91
1 parent 831aacc
commit b5e3b91
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/references/detection/coco_utils.py b/references/detection/coco_utils.py
@@ -6,7 +6,7 @@
 
 from pycocotools import mask as coco_mask
 from pycocotools.coco import COCO
-from torchvision.datapoints import wrap_dataset_for_transforms_v2
+from torchvision.datasets import wrap_dataset_for_transforms_v2
 
 
 def convert_coco_poly_to_mask(segmentations, height, width):

diff --git a/references/detection/presets.py b/references/detection/presets.py
@@ -2,6 +2,8 @@
 
 import torch
 import transforms as reference_transforms
+import torchvision
+torchvision.disable_beta_transforms_warning()
 from torchvision import datapoints
 import torchvision.transforms.v2 as T
 

diff --git a/references/detection/train.py b/references/detection/train.py
@@ -164,6 +164,7 @@ def get_args_parser(add_help=True):
 
 
 def main(args):
+
     if args.output_dir:
         utils.mkdir(args.output_dir)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -164,6 +164,7 @@ def get_args_parser(add_help=True):


		def main(args):

		if args.output_dir:
		utils.mkdir(args.output_dir)

Expand Down