def main(args=None):
"""
Execute the main body of the script.
Parameters
----------
args : list, optional
Command-line arguments. If unspecified, `sys.argv[1:]` is used.
"""
parser = argparse.ArgumentParser(description='Load a YAML file without '
'performing any training.')
parser.add_argument('yaml_file', type=argparse.FileType('r'),
help='The YAML file to load.')
parser.add_argument('-N', '--no-instantiate',
action='store_const', default=False, const=True,
help='Only verify that the YAML parses correctly '
'but do not attempt to instantiate the objects. '
'This might be used as a quick sanity check if '
'checking a file that requires a GPU in an '
'environment that lacks one (e.g. a cluster '
'head node)')
args = parser.parse_args(args=args)
name = args.yaml_file.name
initialize()
if args.no_instantiate:
yaml_load(args.yaml_file)
print("Successfully parsed %s (but objects not instantiated)." % name)
else:
load(args.yaml_file)
print("Successfully parsed and loaded %s." % name)
def test_duplicate_keywords():
"""
Tests whether there are doublicate keywords in the yaml
"""
initialize()
yamlfile = """{
"model": !obj:pylearn2.models.mlp.MLP {
"layers": [
!obj:pylearn2.models.mlp.Sigmoid {
"layer_name": 'h0',
"dim": 20,
"sparse_init": 15,
}],
"nvis": 784,
"nvis": 384,
}
}"""
try:
load(yamlfile)
except yaml.constructor.ConstructorError as e:
message = str(e)
assert message.endswith("found duplicate key (nvis)")
pass
except Exception as e:
error_msg = "Got the unexpected error: %s" % (e)
raise TypeError(error_msg)
def main(args=None):
"""
Execute the main body of the script.
Parameters
----------
args : list, optional
Command-line arguments. If unspecified, `sys.argv[1:]` is used.
"""
parser = argparse.ArgumentParser(description='Load a YAML file without '
'performing any training.')
parser.add_argument('yaml_file',
type=argparse.FileType('r'),
help='The YAML file to load.')
parser.add_argument('-N',
'--no-instantiate',
action='store_const',
default=False,
const=True,
help='Only verify that the YAML parses correctly '
'but do not attempt to instantiate the objects. '
'This might be used as a quick sanity check if '
'checking a file that requires a GPU in an '
'environment that lacks one (e.g. a cluster '
'head node)')
args = parser.parse_args(args=args)
name = args.yaml_file.name
initialize()
if args.no_instantiate:
yaml_load(args.yaml_file)
print "Successfully parsed %s (but objects not instantiated)." % name
else:
load(args.yaml_file)
print "Successfully parsed and loaded %s." % name
def test_duplicate_keywords():
"""
Tests whether there are doublicate keywords in the yaml
"""
initialize()
yamlfile = """{
"model": !obj:pylearn2.models.mlp.MLP {
"layers": [
!obj:pylearn2.models.mlp.Sigmoid {
"layer_name": 'h0',
"dim": 20,
"sparse_init": 15,
}],
"nvis": 784,
"nvis": 384,
}
}"""
try:
load(yamlfile)
except yaml.constructor.ConstructorError as e:
message = str(e)
assert message.endswith("found duplicate key (nvis)")
pass
except Exception as e:
error_msg = "Got the unexpected error: %s" % (e)
raise TypeError(error_msg)
def test_parse_null_as_none():
"""
Tests whether None may be passed via yaml kwarg null.
"""
initialize()
yamlfile = """{
"model": !obj:pylearn2.models.autoencoder.Autoencoder {
"nvis" : 1024,
"nhid" : 64,
"act_enc" : Null,
"act_dec" : null
}
}"""
load(yamlfile)
def test_parse_null_as_none():
"""
Tests whether None may be passed via yaml kwarg null.
"""
initialize()
yamlfile = """{
"model": !obj:pylearn2.models.autoencoder.Autoencoder {
"nvis" : 1024,
"nhid" : 64,
"act_enc" : Null,
"act_dec" : null
}
}"""
load(yamlfile)
def test_duplicate_keywords_2():
"""
Tests whether duplicate keywords as independent parameters works fine.
"""
initialize()
yamlfile = """{
"model": !obj:pylearn2.models.rbm.GaussianBinaryRBM {
"vis_space" : &vis_space !obj:pylearn2.space.Conv2DSpace {
"shape" : [32,32],
"num_channels" : 3
},
"hid_space" : &hid_space !obj:pylearn2.space.Conv2DSpace {
"shape" : [27,27],
"num_channels" : 10
},
"transformer" :
!obj:pylearn2.linear.conv2d.make_random_conv2D {
"irange" : .05,
"input_space" : *vis_space,
"output_space" : *hid_space,
"kernel_shape" : [6,6],
"batch_size" : &batch_size 5
},
"energy_function_class" :
!obj:pylearn2.energy_functions.rbm_energy.grbm_type_1 {},
"learn_sigma" : True,
"init_sigma" : .3333,
"init_bias_hid" : -2.,
"mean_vis" : False,
"sigma_lr_scale" : 1e-3
}
}"""
loaded = load(yamlfile)
def test_parse_null_as_none():
"""
Tests whether None may be passed via yaml kwarg null.
"""
initialize()
yamlfile = """{
