def METHOD_NAME(self):
    valid, _ = testing_utils.train_model(
        dict(
            task='integration_tests:candidate',
            model='transformer/ranker',
            optimizer='adam',
            fp16=True,
            fp16_impl='mem_efficient',
            learningrate=7e-3,
            batchsize=32,
            num_epochs=0.25,
            n_layers=1,
            n_heads=1,
            ffn_size=32,
            embedding_size=32,
            warmup_updates=1,
            lr_scheduler='invsqrt',
        )
    )
    self.assertGreaterEqual(valid['hits@1'], 0.1)