"vscode:/vscode.git/clone" did not exist on "15e91d721b0a0afe0de0706860a03ae810357dd6"
  • Matthew Yu's avatar
    algorithm · 150db2d1
    Matthew Yu authored
    Summary:
    Pull Request resolved: https://github.com/facebookresearch/d2go/pull/431
    
    Add a generic domain adaptation algorithm. This algorithm:
    * gets domain0 data out of the dataloader
    * runs domain0 data into the model and saves target layer output
    * gets domain1 data of the dataloader
    * runs domain1 data into the model and saves target layer output
    * runs domain adaptation loss on domain0, domain1 outputs
    * combines losses using model training iteration
    
    This diffs adds `get_preprocess_domain0_input` and `get_preprocess_domain1_input` to the distillation helper. These are functions that the user can use to convert the dataloader output to something that will be used by the model (e.g., pull the domain0 or domain1 key out of a dataloader that returns a dict).
    
    Differential Revision: D40970724
    
    fbshipit-source-id: fff050fbe864654fa6cb0df927f6843855ec1c14
    150db2d1
test_modeling_distillation.py 27 KB