Continue work on fine-tuning
Cleanup
Implement weight decay
Generalize use of weight