vis4d.common.release

Convert Vis4D model weights for release.

Functions

main()

Main function.

save_weights_with_hash(state_dict, path, ...)

Saves the model weights and append a 6-digit hash to the filename.
save_weights_with_hash(state_dict, path, filename, digits=6)[source]

Saves the model weights and append a 6-digit hash to the filename.

Parameters:

  • state_dict (dict[str, torch.Tensor]) – The model weights to save.

  • path (str) – The directory path to save the model.

  • filename (str) – The filename to save the model.

  • digits (int, optional) – The number of digits to use for the hash. Defaults to 6.

Return type:

None

main()[source]

Main function.

Return type:

None