Skip to content

Commit 041d38a

Browse files
authored
add a note about the device argument
1 parent 6b62593 commit 041d38a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

FaceParsing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def parse_face(self, images, device=0):
5353
# 14 - neck, 15 - neck_l (necklace), 16 - cloth, 17 - hair, 18 - hat
5454

5555
# move the network to the correct device
56+
# REVIEW this makes it imposible to run on the cpu...
5657
self.net.to('cuda:{}'.format(device))
5758

5859
assert all(im.size[0] == im.size[1] for im in images)

0 commit comments

Comments
 (0)