Solving more gpu memory problems.
Still dealing with gpu memory issues.