• Kai Zhang's avatar
    Support evaluate predictor · 6aec097e
    Kai Zhang authored
    Summary:
    Evaluate the predictor generated by previous step.
    This diff modify the lightning_train_net to reuse the evaluation logic by adding a `predictor_path` param.
    This diff also makes Lightning training backend depends on `cfg.MODEL.DEVICE` so that in evaluate_predictor step, user could set backend by changing model device. This is useful for evaluating int8 quantized model.
    
    Reviewed By: newstzpz
    
    Differential Revision: D27150609
    
    fbshipit-source-id: fb72da3e81db932c0fa479350150720143e09a3e
    6aec097e
lightning_train_net.py 7.55 KB