• Zac Wellmer's avatar
    Resnet transfer learning (#5047) · 7bffd37b
    Zac Wellmer authored
    * warm start a resent with all but the dense layer and only update the final layer weights when fine tuning
    
    * Update README for Transfer Learning
    
    * make lint happy and variable naming error related to scaled gradients
    
    * edit the test cases for cifar10 and imagenet to reflect the default case of no fine tuning
    7bffd37b
resnet_run_loop.py 21.1 KB