Browse Source

Update example.py

pull/3/head
troyyyyy GitHub 2 years ago
parent
commit
9f8589d0cb
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      example.py

+ 1
- 1
example.py View File

@@ -71,7 +71,7 @@ def run_test():
base_model = BasicModel(cls, criterion, optimizer, device, Params(), sign_list, recorder=recorder)
model = MyModel(base_model)

res = framework.train(model, abducer, train_X, train_Y, logic_forward = kb.logic_forward, sample_num = 10000, verbose = 1)
res = framework.train(model, abducer, train_X, train_Y, sample_num = 10000, verbose = 1)
print(res)



Loading…
Cancel
Save