mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-15 06:00:07 +00:00
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
This commit is contained in:
@@ -9,8 +9,7 @@ from ._utils import calc_acc
|
||||
|
||||
|
||||
class Accuracy3D(nn.Module):
|
||||
"""Accuracy for 3D parallelism
|
||||
"""
|
||||
"""Accuracy for 3D parallelism"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
@@ -26,7 +25,7 @@ class Accuracy3D(nn.Module):
|
||||
|
||||
Returns:
|
||||
float: the accuracy of prediction.
|
||||
"""
|
||||
"""
|
||||
with torch.no_grad():
|
||||
targets = split_tensor_3d(targets, 0, self.weight_parallel_mode)
|
||||
targets = split_tensor_3d(targets, 0, self.input_parallel_mode)
|
||||
|
Reference in New Issue
Block a user