Spaces:
Runtime error
Runtime error
Update config.py
Browse files
config.py
CHANGED
|
@@ -94,34 +94,7 @@ config = Config()
|
|
| 94 |
###################################### arguments ######################################
|
| 95 |
def parseArgs():
|
| 96 |
parser = argparse.ArgumentParser(fromfile_prefix_chars = "@")
|
| 97 |
-
|
| 98 |
-
parallel=True,
|
| 99 |
-
evalTrain=True,
|
| 100 |
-
retainVal=True,
|
| 101 |
-
useEMA=True,
|
| 102 |
-
lrReduce=True,
|
| 103 |
-
adam=True,
|
| 104 |
-
clip=True,
|
| 105 |
-
memoryVariationalDropout=True,
|
| 106 |
-
relu='ELU',
|
| 107 |
-
encBi=True,
|
| 108 |
-
wrdEmbRandom=True,
|
| 109 |
-
wrdEmbUniform=True,
|
| 110 |
-
outQuestion=True,
|
| 111 |
-
initCtrl='Q',
|
| 112 |
-
controlContextual=True,
|
| 113 |
-
controlInputUnshared=True,
|
| 114 |
-
readProjInputs=True,
|
| 115 |
-
readMemConcatKB=True,
|
| 116 |
-
readMemConcatProj=True,
|
| 117 |
-
readMemProj=True,
|
| 118 |
-
readCtrl=True,
|
| 119 |
-
writeMemProj=True,
|
| 120 |
-
restore=True,
|
| 121 |
-
expName='PDF_exp_extra',
|
| 122 |
-
netLength=16
|
| 123 |
-
)
|
| 124 |
-
|
| 125 |
################ systems
|
| 126 |
|
| 127 |
#custom args
|
|
@@ -453,7 +426,33 @@ def parseArgs():
|
|
| 453 |
# parser.add_argument("--gumbelSoftmaxBoth", action = "store_true", help = "use softmax for training and testing") #
|
| 454 |
# parser.add_argument("--gumbelArgmaxBoth", action = "store_true", help = "use argmax for training and testing") #
|
| 455 |
|
| 456 |
-
parser.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 457 |
|
| 458 |
###################################### dataset configuration ######################################
|
| 459 |
|
|
|
|
| 94 |
###################################### arguments ######################################
|
| 95 |
def parseArgs():
|
| 96 |
parser = argparse.ArgumentParser(fromfile_prefix_chars = "@")
|
| 97 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 98 |
################ systems
|
| 99 |
|
| 100 |
#custom args
|
|
|
|
| 426 |
# parser.add_argument("--gumbelSoftmaxBoth", action = "store_true", help = "use softmax for training and testing") #
|
| 427 |
# parser.add_argument("--gumbelArgmaxBoth", action = "store_true", help = "use argmax for training and testing") #
|
| 428 |
|
| 429 |
+
parser.set_defaults(
|
| 430 |
+
parallel=True,
|
| 431 |
+
evalTrain=True,
|
| 432 |
+
retainVal=True,
|
| 433 |
+
useEMA=True,
|
| 434 |
+
lrReduce=True,
|
| 435 |
+
adam=True,
|
| 436 |
+
clip=True,
|
| 437 |
+
memoryVariationalDropout=True,
|
| 438 |
+
relu='ELU',
|
| 439 |
+
encBi=True,
|
| 440 |
+
wrdEmbRandom=True,
|
| 441 |
+
wrdEmbUniform=True,
|
| 442 |
+
outQuestion=True,
|
| 443 |
+
initCtrl='Q',
|
| 444 |
+
controlContextual=True,
|
| 445 |
+
controlInputUnshared=True,
|
| 446 |
+
readProjInputs=True,
|
| 447 |
+
readMemConcatKB=True,
|
| 448 |
+
readMemConcatProj=True,
|
| 449 |
+
readMemProj=True,
|
| 450 |
+
readCtrl=True,
|
| 451 |
+
writeMemProj=True,
|
| 452 |
+
restore=True,
|
| 453 |
+
expName='PDF_exp_extra',
|
| 454 |
+
netLength=16
|
| 455 |
+
)
|
| 456 |
|
| 457 |
###################################### dataset configuration ######################################
|
| 458 |
|