train_output_network.py

import torch




def main():
	pass





def load_datasets(data_dir: str = "../data"):
    train_data = torch.load(os.path.join(data_dir, "train.pt"))
    val_data = torch.load(os.path.join(data_dir, "test.pt"))
    return train_data, val_data