Closed chschroeder closed 2 years ago
Passing a scheduler to TransformerBasedClassification.fit() results in AttributeError: 'str' object has no attribute 'step'.
TransformerBasedClassification.fit()
AttributeError: 'str' object has no attribute 'step'
clf.fit(dataset, optimizer=optimizer, scheduler=scheduler)
where clf is an instance of TransformerBasedClassification and optimizer and scheduler are not None.
TransformerBasedClassification
The given optimizer and scheduler should be used.
small-text version: 1.0.0 small-text integrations (e.g., transformers): transformers
The error is here
Bug description
Passing a scheduler to
TransformerBasedClassification.fit()results inAttributeError: 'str' object has no attribute 'step'.Steps to reproduce
where clf is an instance of
TransformerBasedClassificationand optimizer and scheduler are not None.Expected behavior
The given optimizer and scheduler should be used.
Environment:
small-text version: 1.0.0 small-text integrations (e.g., transformers): transformers
Addition information
The error is here