mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-30 13:40:46 +08:00
working on trainer docs
This commit is contained in:
1 parent
80454d3635
commit
1b12cb83dc
1 file changed
+4
-1
@@ -355,7 +355,10 @@ class Trainer(TrainerIOMixin,
|
||||
# if checkpoint callback used, then overrides the weights path
|
||||
# **NOTE: this saves weights to some/path NOT my/path
|
||||
checkpoint_callback = ModelCheckpoint(filepath='some/path')
|
||||
trainer = Trainer(checkpoint_callback=checkpoint_callback, weights_save_path='my/path')
|
||||
trainer = Trainer(
|
||||
checkpoint_callback=checkpoint_callback,
|
||||
weights_save_path='my/path'
|
||||
)
|
||||
|
||||
# :param str amp_level: Check nvidia docs for level
|
||||
"""
|
||||
|
||||
Reference in new issue
Block a user