File our_train_B.py

src/script/our_train_B.py:None–None

import os
import warnings
import argparse
import wandb