Commit 6253d3cb authored by miaecle's avatar miaecle
Browse files

fix failure

parent 3acd609c
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -116,7 +116,7 @@ class ANIRegression(TensorGraph):
               n_tasks,
               max_atoms,
               exp_loss=False,
               activation_fn='relu',
               activation_fn='ani',
               layer_structures=[128, 64],
               atom_number_cases=[1, 6, 7, 8, 16],
               **kwargs):
+1 −1
Original line number Diff line number Diff line
@@ -28,7 +28,7 @@ n_distance = 51
distance_min = -1.
distance_max = 9.2
n_hidden = 15
rate = 0.0005
rate = 0.001

model = dc.models.DTNNModel(
    len(tasks),