Multi-GPU update with custom collate function to allow variable size target vector per image without needing to pad targets. |
||
|---|---|---|
| .. | ||
| datasets.py | ||
| gcp.sh | ||
| parse_config.py | ||
| torch_utils.py | ||
| utils.py | ||
Multi-GPU update with custom collate function to allow variable size target vector per image without needing to pad targets. |
||
|---|---|---|
| .. | ||
| datasets.py | ||
| gcp.sh | ||
| parse_config.py | ||
| torch_utils.py | ||
| utils.py | ||