/usr/local/lib/python3.8/dist-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx (Triggered internally at /pytorch/c10/cuda/CUDAFunctions.cpp:100.) return torch._C._cuda_getDeviceCount() > 0
synth_dbunch
[source]
synth_dbunch
(a
=2
,b
=3
,bs
=16
,n_train
=10
,n_valid
=2
,cuda
=False
)
synth_learner
[source]
synth_learner
(n_trn
=10
,n_val
=2
,cuda
=False
,lr
=0.001
,data
=None
,model
=None
,loss_func
=None
,opt_func
=Adam
,splitter
=trainable_params
,cbs
=None
,metrics
=None
,path
=None
,model_dir
='models'
,wd
=None
,wd_bn_bias
=False
,train_bn
=True
,moms
=(0.95, 0.85, 0.95)
)
class
VerboseCallback
[source]
VerboseCallback
(after_create
=None
,before_fit
=None
,before_epoch
=None
,before_train
=None
,before_batch
=None
,after_pred
=None
,after_loss
=None
,before_backward
=None
,before_step
=None
,after_cancel_step
=None
,after_step
=None
,after_cancel_batch
=None
,after_batch
=None
,after_cancel_train
=None
,after_train
=None
,before_validate
=None
,after_cancel_validate
=None
,after_validate
=None
,after_cancel_epoch
=None
,after_epoch
=None
,after_cancel_fit
=None
,after_fit
=None
) ::Callback
Callback that prints the name of each event called
get_env
[source]
get_env
(name
)
Return env var value if it's defined and not an empty string, or return Unknown
try_import
[source]
try_import
(module
)
Try to import module
. Returns module's object on success, None on failure
nvidia_smi
[source]
nvidia_smi
(cmd
='nvidia-smi'
)
res = nvidia_smi()
nvidia_mem
[source]
nvidia_mem
()
nvidia_mem()
['48600', '7982']