def _write(d):
filing = Filing(
url=Filing.HTTP_ROOT + d['folder']
)
path = filing.md_path()
print 'Writing:', path
text = d['text']
if not d['text_file']:
text = clean(text)
with open(path, 'w') as f:
f.write(text)
force_cpu=False).astype('float32')
logit = fluid.layers.transpose(logit, [0, 2, 3, 1])
logit = fluid.layers.reshape(logit, [-1, num_classes])
label = fluid.layers.reshape(label, [-1, 1])
label = fluid.layers.cast(label, 'int64')
label_nignore = fluid.layers.reshape(label_nignore, [-1, 1])
loss = fluid.layers.softmax_with_cross_entropy(logit, label, ignore_index=255, numeric_stable_mode=True)
label_nignore.stop_gradient = True
label.stop_gradient = True
return loss, label_nignore
args = parser.parse_args()
utility.print_arguments(args)
models.clean()
models.bn_momentum = 0.9997
models.dropout_keep_prop = 0.9
models.label_number = args.num_classes
models.default_norm_type = args.norm_type
deeplabv3p = models.deeplabv3p
sp = fluid.Program()
tp = fluid.Program()
# only for ce
if args.enable_ce:
SEED = 102
sp.random_seed = SEED
tp.random_seed = SEED