mirror of
https://github.com/facebookresearch/blt.git
synced 2025-01-19 00:47:44 +00:00
remove unused argument
This commit is contained in:
parent
420326184a
commit
6129756e10
|
@ -199,9 +199,6 @@ class LocalModelBase(nn.Module):
|
||||||
class LocalEncoder(LocalModelBase):
|
class LocalEncoder(LocalModelBase):
|
||||||
def __init__(self, args: LocalModelArgs):
|
def __init__(self, args: LocalModelArgs):
|
||||||
super().__init__(args)
|
super().__init__(args)
|
||||||
self.output_proj = (
|
|
||||||
args.patching_mode in ["entropy", "probmax"]
|
|
||||||
) and args.entropy_model_checkpoint_dir is None
|
|
||||||
|
|
||||||
self.apply_transformer = args.use_local_encoder_transformer
|
self.apply_transformer = args.use_local_encoder_transformer
|
||||||
self.downsampling_by_pooling = args.downsampling_by_pooling
|
self.downsampling_by_pooling = args.downsampling_by_pooling
|
||||||
|
|
Loading…
Reference in a new issue