def __init__(self, net=pb.NetworkFormat.NETWORK_SE_WITH_HEADFORMAT, input=pb.NetworkFormat.INPUT_CLASSICAL_112_PLANE, value=pb.NetworkFormat.VALUE_CLASSICAL, policy=pb.NetworkFormat.POLICY_CLASSICAL, moves_left=pb.NetworkFormat.MOVES_LEFT_V1): if net == pb.NetworkFormat.NETWORK_SE: net = pb.NetworkFormat.NETWORK_SE_WITH_HEADFORMAT if net == pb.NetworkFormat.NETWORK_CLASSICAL: net = pb.NetworkFormat.NETWORK_CLASSICAL_WITH_HEADFORMAT self.pb = pb.Net() self.pb.magic = WEIGHTS_MAGIC self.pb.min_version.major = LC0_MAJOR self.pb.min_version.minor = LC0_MINOR self.pb.min_version.patch = LC0_PATCH self.pb.format.weights_encoding = pb.Format.LINEAR16 self.weights = [] self.set_networkformat(net) self.pb.format.network_format.input = input self.set_policyformat(policy) self.set_valueformat(value) self.set_movesleftformat(moves_left) self.set_defaultactivation(pb.NetworkFormat.DEFAULT_ACTIVATION_RELU)
def __init__(self): self.pb = pb.Net() self.pb.magic = WEIGHTS_MAGIC self.pb.min_version.major = LC0_MAJOR self.pb.min_version.minor = LC0_MINOR self.pb.min_version.patch = LC0_PATCH self.pb.format.weights_encoding = pb.Format.LINEAR16 self.weights = []
def __init__(self, net=pb.NetworkFormat.NETWORK_SE_WITH_HEADFORMAT, input=pb.NetworkFormat.INPUT_CLASSICAL_112_PLANE, value=pb.NetworkFormat.VALUE_CLASSICAL, policy=pb.NetworkFormat.POLICY_CLASSICAL): if net == pb.NetworkFormat.NETWORK_SE: net = pb.NetworkFormat.NETWORK_SE_WITH_HEADFORMAT if net == pb.NetworkFormat.NETWORK_CLASSICAL: net = pb.NetworkFormat.NETWORK_CLASSICAL_WITH_HEADFORMAT self.pb = pb.Net() self.pb.magic = WEIGHTS_MAGIC self.pb.min_version.major = LC0_MAJOR self.pb.min_version.minor = LC0_MINOR self.pb.min_version.patch = LC0_PATCH self.pb.format.weights_encoding = pb.Format.LINEAR16 self.weights = [] self.set_networkformat(net) self.pb.format.network_format.input = input self.set_policyformat(policy) self.set_valueformat(value)