Update train.py

train.py

@@ -279,7 +279,7 @@ if __name__ == '__main__':
         
     augmentations = get_transforms()
     
-    train_dataloader, ft_dataloader = get_dataloaders(args.path, augmentations, config['batch_size'], args.fine_tuning, config['number_of_mults'])
+    train_dataloader, ft_dataloader = get_dataloaders(args.path, augmentations, config['batch_size'], args.fine_tuning)
     
     colorizer, discriminator, content = get_models(device)
     set_weights(colorizer, discriminator)