modifié : config/config.py

	modifié :         dataset/dataset.py

config/config.py

@@ -5,7 +5,7 @@ def load_args():
     parser = argparse.ArgumentParser()
     parser.add_argument('--epoches', type=int, default=20)
     parser.add_argument('--eval_inter', type=int, default=1)
-    parser.add_argument('--augment_args', nargs = '+', type = float, default = [0,0,0,0.1,0.1,0.1,0.1])
+    parser.add_argument('--augment_args', nargs = '+', type = float, default = [0,0,0.5,0.1,0.1,0.,7.5])
     parser.add_argument('--noise_threshold', type=int, default=0)
     parser.add_argument('--lr', type=float, default=0.001)
     parser.add_argument('--optim', type = str, default = "Adam")

dataset/dataset.py

@@ -221,7 +221,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
     train_transform = transforms.Compose(
         [#Random_erasing(args.augment_args[0], args.augment_args[3]),
          #Random_int_noise(args.augment_args[1], args.augment_args[4]),
-         #Random_shift_rt(args.augment_args[2], args.augment_args[5], args.augment_args[6]),
+         Random_shift_rt(args.augment_args[2], args.augment_args[5], args.augment_args[6]),
          transforms.Resize((224, 224)),
          Threshold_noise(args.noise_threshold),
          Log_normalisation(),
@@ -254,7 +254,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
         shuffle=shuffle,
         num_workers=8,
         collate_fn=None,
-        pin_memory=True,
+        pin_memory=False,
     )
     data_loader_test = data.DataLoader(
         dataset=val_dataset,
@@ -262,7 +262,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
         shuffle=shuffle,
         num_workers=8,
         collate_fn=None,
-        pin_memory=True,
+        pin_memory=False,
     )
 
     return data_loader_train, data_loader_test