fbcbb710c465da13ed6705fb8d904fe2795b246a,luminoth/models/base/truncated_base_network.py,TruncatedBaseNetwork,get_trainable_vars,#TruncatedBaseNetwork#,81

Before Change

        for index in scope_var_index_iter:
            pass

        if index is None:
            raise ValueError(
                ""{}" is an invalid value of endpoint for this "
                "architecture.".format(self._endpoint)
            )

        return all_trainable[:index + 1]

    def _get_endpoint(self, endpoints):
        

After Change

            // TODO: Tail should be configurable too, to avoid hard-coding
            // the trainable portion to `block4` and allow using something in
            // block4 as endpoint.
            var_iter = enumerate(v.name for v in all_trainable)
            try:
                index = next(i for i, name in var_iter if "block4" in name)
            except StopIteration:
                raise ValueError(
                    ""block4" not present in the trainable vars retrieved "

Instances