Increase size of loss_scale_test to fix nightly build
PiperOrigin-RevId: 241830333
This commit is contained in:
parent
24405dd523
commit
cc686693f3
@ -126,7 +126,7 @@ py_library(
|
|||||||
|
|
||||||
py_test(
|
py_test(
|
||||||
name = "loss_scale_test",
|
name = "loss_scale_test",
|
||||||
size = "small",
|
size = "medium",
|
||||||
srcs = ["loss_scale_test.py"],
|
srcs = ["loss_scale_test.py"],
|
||||||
deps = [
|
deps = [
|
||||||
":loss_scale",
|
":loss_scale",
|
||||||
|
Loading…
Reference in New Issue
Block a user