Fixed some changes that weren't carried over.
Added a file to use the learned weights from Tian's code.