|
|
2025-08-15 05:14:18,140 INFO MainThread:4156 [wandb_run.py:_config_callback():1380] config_cb None None {'out': 'outputs/ssl_dino/swin_tiny', 'data': 'data/kyucapsule', 'model': 'SwinTransformer', 'method': 'dino', 'method_args': {'hidden_dim': 2048, 'bottleneck_dim': 256, 'output_dim': 1024, 'student_freeze_last_layer_epochs': 1, 'batch_norm': False, 'norm_last_layer': True, 'teacher_temp': 0.02, 'warmup_teacher_temp': 0.02, 'warmup_teacher_temp_epochs': 30, 'student_temp': 0.1, 'center_momentum': 0.9, 'momentum_start': 0.99, 'momentum_end': 1.0, 'weight_decay_start': 0.0001, 'weight_decay_end': 0.0001}, 'embed_dim': None, 'epochs': 300, 'batch_size': 32, 'num_workers': 64, 'devices': 1, 'num_nodes': 1, 'resume': True, 'checkpoint': None, 'overwrite': True, 'accelerator': 'CUDAAccelerator', 'strategy': 'SingleDeviceStrategy', 'precision': '32-true', 'seed': 0, 'loggers': {'jsonl': {'flush_logs_every_n_steps': 100}, 'tensorboard': {'name': '', 'version': '', 'log_graph': False, 'default_hp_metric': True, 'prefix': '', 'sub_dir': None}, 'wandb': {'name': None, 'version': None, 'offline': False, 'anonymous': None, 'project': 'ent-endoscopy-ssl', 'log_model': False, 'prefix': '', 'checkpoint_name': None}}, 'callbacks': {'learning_rate_monitor': {}, 'device_stats_monitor': {}, 'early_stopping': {'monitor': 'train_loss', 'patience': 1000000000000, 'check_finite': True}, 'model_checkpoint': {'save_last': True, 'enable_version_counter': False, 'save_top_k': 1, 'every_n_epochs': None}}, 'optim': 'sgd', 'optim_args': {'lr': 0.03, 'momentum': 0.9, 'weight_decay': 0.0001}, 'transform_args': {'image_size': [224, 224], 'random_resize': {'min_scale': 0.14, 'max_scale': 1.0}, 'random_flip': {'horizontal_prob': 0.5, 'vertical_prob': 0.0}, 'random_rotation': None, 'color_jitter': {'prob': 0.8, 'strength': 0.5, 'brightness': 0.8, 'contrast': 0.8, 'saturation': 0.4, 'hue': 0.2}, 'random_gray_scale': 0.2, 'normalize': {'mean': [0.485, 0.456, 0.406], 'std': [0.229, 0.224, 0.225]}, 'gaussian_blur': {'prob': 1.0, 'sigmas': [0.1, 2.0], 'blur_limit': 0}, 'solarize': None, 'global_view_1': {'gaussian_blur': {'prob': 0.1, 'sigmas': [0.1, 2.0], 'blur_limit': 0}, 'solarize': {'prob': 0.2, 'threshold': 0.5}}, 'local_view': {'num_views': 0, 'view_size': [96, 96], 'random_resize': {'min_scale': 0.05, 'max_scale': 0.14}, 'gaussian_blur': {'prob': 0.5, 'sigmas': [0.1, 2.0], 'blur_limit': 0}}}, 'loader_args': None, 'trainer_args': None, 'model_args': None} |