Commit 22b36bc3 authored by Shreyan Chowdhury's avatar Shreyan Chowdhury
Browse files

fscore with 0.5 threshold

parent f48eb007
......@@ -16,7 +16,7 @@ def run():
if USE_GPU:
trainer = Trainer(gpus=[0], distributed_backend='ddp',
experiment=exp, max_nb_epochs=500, train_percent_check=1.0,
experiment=exp, max_nb_epochs=1, train_percent_check=0.1,
fast_dev_run=False)
else:
trainer = Trainer(experiment=exp, max_nb_epochs=1, train_percent_check=0.1,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment