|
|
@@ -51,13 +51,13 @@ if __name__ == '__main__':
|
|
|
devstone_model = None
|
|
|
start_time = time.time()
|
|
|
if args.model_type == "LI":
|
|
|
- devstone_model = LI("LI_root", args.depth, args.width, args.int_cycles, args.ext_cycles)
|
|
|
+ devstone_model = LI("LI_root", args.depth, args.width, args.int_cycles, args.ext_cycles, prep_time=1)
|
|
|
elif args.model_type == "HI":
|
|
|
- devstone_model = HI("HI_root", args.depth, args.width, args.int_cycles, args.ext_cycles)
|
|
|
+ devstone_model = HI("HI_root", args.depth, args.width, args.int_cycles, args.ext_cycles, prep_time=1)
|
|
|
elif args.model_type == "HO":
|
|
|
- devstone_model = HO("HO_root", args.depth, args.width, args.int_cycles, args.ext_cycles)
|
|
|
+ devstone_model = HO("HO_root", args.depth, args.width, args.int_cycles, args.ext_cycles, prep_time=1)
|
|
|
elif args.model_type == "HOmod":
|
|
|
- devstone_model = HOmod("HOmod_root", args.depth, args.width, args.int_cycles, args.ext_cycles)
|
|
|
+ devstone_model = HOmod("HOmod_root", args.depth, args.width, args.int_cycles, args.ext_cycles, prep_time=1)
|
|
|
|
|
|
env = DEVStoneEnvironment("DEVStoneEnvironment", devstone_model)
|
|
|
model_created_time = time.time()
|