Summary: 4 instances, 3 unique

Text	Count
# TODO: make it with torch instead of numpy	1
# TODO: make it with torch instead of numpy	1
# TODO:change this part	2