Merge branch 'main' into dockerfile

Author: vfdev-5

.github/workflows/ci.yml

@@ -93,7 +93,7 @@ jobs:
 
       - uses: actions/setup-node@v3
         with:
-          node-version: 16
+          node-version: 18
 
       - uses: actions/cache@v3
         with:

scripts/run_code_style.sh

@@ -8,5 +8,5 @@ if [ $1 == "lint" ]; then
 elif [ $1 == "fmt" ]; then
     ufmt format .
 elif [ $1 == "install" ]; then
-    pip install --upgrade "black==21.12b0" "usort==1.0.2" "ufmt==1.3.2" flake8
+    pip install --upgrade "black==23.3.0" "usort==1.0.6" "ufmt==2.1.0" flake8
 fi

src/templates/template-text-classification/main.py

@@ -19,7 +19,6 @@
 
 
 def run(local_rank: int, config: Any):
-
     # make a certain seed
     rank = idist.get_rank()
     manual_seed(config.seed + rank)

src/templates/template-text-classification/models.py

@@ -10,7 +10,6 @@ def __init__(self, model_name, model_dir, dropout, n_fc, n_classes):
         self.classifier = nn.Linear(n_fc, n_classes)
 
     def forward(self, ids, mask, token_type_ids):
-
         hidden_output, pooled_output = self.transformer(
             ids,
             attention_mask=mask,

src/templates/template-text-classification/trainers.py

@@ -18,7 +18,6 @@ def setup_trainer(
     device: Union[str, torch.device],
     train_sampler: Sampler,
 ) -> Union[Engine, DeterministicEngine]:
-
     scaler = GradScaler(enabled=config.use_amp)
 
     def train_function(engine: Union[Engine, DeterministicEngine], batch: Any):

src/templates/template-vision-classification/main.py

@@ -14,7 +14,6 @@
 
 
 def run(local_rank: int, config: Any):
-
     # make a certain seed
     rank = idist.get_rank()
     manual_seed(config.seed + rank)

src/templates/template-vision-dcgan/main.py

@@ -18,7 +18,6 @@
 
 
 def run(local_rank: int, config: Any):
-
     # make a certain seed
     rank = idist.get_rank()
     manual_seed(config.seed + rank)

src/templates/template-vision-dcgan/trainers.py

@@ -19,7 +19,6 @@ def setup_trainer(
     device: Union[str, torch.device],
     train_sampler: Sampler,
 ) -> Union[Engine, DeterministicEngine]:
-
     ws = idist.get_world_size()
 
     real_labels = torch.ones(config.train_batch_size // ws, device=device)
@@ -111,7 +110,6 @@ def setup_evaluator(
     loss_fn: Module,
     device: Union[str, torch.device],
 ) -> Engine:
-
     ws = idist.get_world_size()
 
     real_labels = torch.ones(config.eval_batch_size // ws, device=device)

src/templates/template-vision-segmentation/data.py

@@ -28,7 +28,6 @@ def __getitem__(self, index):
 
 
 class VOCSegmentationPIL(VOCSegmentation):
-
     target_names = [
         "background",
         "aeroplane",

src/templates/template-vision-segmentation/main.py

@@ -18,7 +18,6 @@
 
 
 def run(local_rank: int, config: Any):
-
     # make a certain seed
     rank = idist.get_rank()
     manual_seed(config.seed + rank)