.drone.yml

@@ -22,4 +22,4 @@ steps:
       - mkdir -p data
       - $mount_command
       - ls data/
-      - CI=False python resnet_finetune_vggface.py
+      - CI=True python resnet_finetune_vggface.py

resnet_finetune_vggface.py

@@ -111,7 +111,7 @@ print()
 
 tune = False
 for name, layer in model.named_parameters():
-    if 'layer1' in name:
+    if 'layer3' in name:
         tune = True
     layer.requires_grad = tune
 
@@ -157,7 +157,7 @@ for epoch in range(epochs):
     model_gpu.train()
     makedirs(join(save_dir, 'pallets', str(epoch)), exist_ok=True)
 
-    for count, (images, labels) in enumerate(tqdm(dataloader['train'], disable=CI, mininterval=20)):
+    for count, (images, labels) in enumerate(tqdm(dataloader['train'], disable=CI)):
         if count == 1:
             image_pallets = plot_dataset(dataloader=(images, labels), col_len=6,
                                          label_text=image_folder['train'].classes)