Niksa Praljak
update all README.md
f782d11
|
raw
history blame
882 Bytes

Downloading Pre-trained Weights

The Google Drive link for downloading the Facilitator pre-trained weights will be added here soon.


File Details

  • File Name: Facilitator pre-trained weights (TBD).
  • Description: Pre-trained weights for the Facilitator model.

Usage

Once available, the pre-trained weights can be loaded as follows:

import torch
model = YourFacilitatorModel()  # Replace with your model class
model.load_state_dict(torch.load("weights/Facilitator/Facilitator_weights.bin", map_location="cpu"))
model.eval()