We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1432c09 commit 1b88f57Copy full SHA for 1b88f57
torchtitan/experiments/vlm/__init__.py
@@ -7,6 +7,7 @@
7
from dataclasses import fields
8
from typing import Any
9
10
+from torchtitan.components.loss import build_cross_entropy_loss
11
from torchtitan.components.lr_scheduler import build_lr_schedulers
12
from torchtitan.components.optimizer import build_optimizers
13
from torchtitan.components.tokenizer import build_hf_tokenizer
0 commit comments