more cleanup

tasks/vision/finetune_utils.py

@@ -199,7 +199,6 @@ def _train(
                     process_non_loss_data_func,
                     False,
                 )
-                end_of_epoch_callback(model, epoch)
 
         # Callback at the end of each epoch.
         if end_of_epoch_callback is not None:

tasks/vision/segmentation/seg_heads.py

@@ -12,9 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
-"""Vision Transformer(VIT) model."""
-
 import math
 import einops
 import torch

tasks/vision/segmentation/seg_models.py

@@ -12,9 +12,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
-"""Vision Transformer(VIT) model."""
-
 import math
 import einops
 import torch