Remove unnecessary configs

release/0.11.0
wangmeng28 7 years ago
parent 28c98103de
commit f504c8a83d

@ -2371,10 +2371,8 @@ void testFactorizationMachineLayer(InputType type, bool useGpu) {
TEST(Layer, FactorizationMachineLayer) { TEST(Layer, FactorizationMachineLayer) {
testFactorizationMachineLayer(INPUT_DATA, false); testFactorizationMachineLayer(INPUT_DATA, false);
testFactorizationMachineLayer(INPUT_SPARSE_FLOAT_VALUE_DATA, false);
#ifdef PADDLE_WITH_CUDA
testFactorizationMachineLayer(INPUT_DATA, true); testFactorizationMachineLayer(INPUT_DATA, true);
#endif testFactorizationMachineLayer(INPUT_SPARSE_FLOAT_VALUE_DATA, false);
} }
int main(int argc, char** argv) { int main(int argc, char** argv) {

@ -1,7 +1,5 @@
from paddle.trainer_config_helpers import * from paddle.trainer_config_helpers import *
settings(batch_size=1000, learning_rate=1e-5)
data = data_layer(name='data', size=1024) data = data_layer(name='data', size=1024)
fm = factorization_machine(input=data, factor_size=10) fm = factorization_machine(input=data, factor_size=10)

Loading…
Cancel
Save