wuchendi
/

MODNet

Image Segmentation

Transformers.js

background-removal

portrait-matting

Model card Files Files and versions

MODNet / test.py

wuchendi's picture

Create test.py

e7a6c95 verified 12 months ago

102 Bytes

	import torch
	x = torch.rand(5, 3)
	print(x)

	print(torch.__version__)
	print(torch.cuda.is_available())