inf-check throws a warning instead of getting terminated when encountering modules do not require gradient

inf-check throws a warning instead of getting terminated when encountering modules do not require gradient #345