Example #1
0
 def tearDown(self):
     registry.unregister("clevr_text_vocab_size")
     registry.unregister("clevr_num_final_outputs")
     registry.unregister("config")
     del self.config
     gc.collect()
Example #2
0
 def tearDown(self):
     if torch.distributed.is_initialized():
         torch.distributed.destroy_process_group()
     del self.trainer
     registry.unregister("distributed")
Example #3
0
def teardown_imports():
    from mmf.common.registry import registry

    registry.unregister("pythia_path")
    registry.unregister("mmf_path")
    registry.unregister("imports_setup")
 def tearDown(self):
     registry.unregister("config")
Example #5
0
 def _sanitize_registry(self):
     registry.mapping["builder_name_mapping"].pop("always_one", None)
     registry.mapping["model_name_mapping"].pop("simple", None)
     registry.unregister("__mmf_user_dir_imported__")