896 half-pretrained!
Browse files
configs/pretrained_config.yaml
CHANGED
|
@@ -18,7 +18,7 @@ parameters:
|
|
| 18 |
REASSEMBLE_OUT_CHANNELS: [768,1024,1536,2048] # Output channels for each decoder stage (DPT-style reassembly)
|
| 19 |
REASSEMBLE_FACTORS: [4.0, 2.0, 1.0, 0.5] # Spatial upsampling factors for each stage
|
| 20 |
READOUT_TYPE: "ignore" # Readout type for DPT decoder ("ignore", "project", etc.)
|
| 21 |
-
FROM_PRETRAINED: "weights/decoder_896.pth" # Path to pretrained decoder weights (optional)
|
| 22 |
USE_BN: False # Use batch normalization in decoder
|
| 23 |
DROPOUT: 0.1 # Dropout rate in decoder layers
|
| 24 |
OUTPUT_IMAGE_SIZE: [896,896] # Output image resolution [height, width]
|
|
|
|
| 18 |
REASSEMBLE_OUT_CHANNELS: [768,1024,1536,2048] # Output channels for each decoder stage (DPT-style reassembly)
|
| 19 |
REASSEMBLE_FACTORS: [4.0, 2.0, 1.0, 0.5] # Spatial upsampling factors for each stage
|
| 20 |
READOUT_TYPE: "ignore" # Readout type for DPT decoder ("ignore", "project", etc.)
|
| 21 |
+
# FROM_PRETRAINED: "weights/decoder_896.pth" # Path to pretrained decoder weights (optional)
|
| 22 |
USE_BN: False # Use batch normalization in decoder
|
| 23 |
DROPOUT: 0.1 # Dropout rate in decoder layers
|
| 24 |
OUTPUT_IMAGE_SIZE: [896,896] # Output image resolution [height, width]
|