def test_VAE_cost():
 yaml_src_path = os.path.join(os.path.dirname(__file__), 'test_vae_cost_vae_criterion.yaml')
 train_object = yaml_parse.load_path(yaml_src_path)
 train_object.main_loop()

def test_VAE_cost ( ) : yaml_src_path = os . path . join ( os . path . dirname ( __file__ ) , 'test_vae_cost_vae_criterion.yaml' ) train_object = yaml_parse . load_path ( yaml_src_path ) train_object . main_loop ( )
vae trains properly with the vae cost .